DeepLung
DeepLung copied to clipboard
question about fuction and parameter
Excuse me? How to use this function,singletest( ) , Does this function detect a CT?in additon,What does this parameter about 'isfeat' mean?Thank you very much!
Could you tell me more about and where you mentioned?
this function: def singletest(data,net,config,splitfun,combinefun,n_per_run,margin = 64,isfeat=False): z, h, w = data.size(2), data.size(3), data.size(4) print(data.size()) data = splitfun(data,config['max_stride'],margin) data = Variable(data.cuda(async = True), volatile = True,requires_grad=False) splitlist = range(0,args.split+1,n_per_run) outputlist = [] featurelist = [] for i in range(len(splitlist)-1): if isfeat: output,feature = net(data[splitlist[i]:splitlist[i+1]]) featurelist.append(feature) else: output = net(data[splitlist[i]:splitlist[i+1]]) output = output.data.cpu().numpy() outputlist.append(output)
output = np.concatenate(outputlist,0)
output = combinefun(output, z / config['stride'], h / config['stride'], w / config['stride'])
if isfeat:
feature = np.concatenate(featurelist,0).transpose([0,2,3,4,1])
feature = combinefun(feature, z / config['stride'], h / config['stride'], w / config['stride'])
return output,feature
else:
return output
I dont know this parameter about 'isfeat' mean. and how to use this function?
May I know which file?