search

wenyalintw / Nodule-CADx

Pulmonary nodules computer-aided diagnosis demo system.
MIT License
20 stars 5 forks source link

a problrm in nodule_net.py #3

Open huangguoqing1111 opened 1 year ago

huangguoqing1111 commented 1 year ago

File "D:\HGQ_DL\Nodule-CADx\src\net\nodule_net.py", line 420, in forward self.mask_probs = data_parallel(self.mask_head, (torch.from_numpy(self.crop_boxes).cuda(), features)) File "D:\anaconda\envs\Stest\lib\site-packages\torch\nn\parallel\data_parallel.py", line 231, in data_parallel return module(*inputs[0], *module_kwargs[0]) File "D:\anaconda\envs\Stest\lib\site-packages\torch\nn\modules\module.py", line 1190, in _call_impl return forward_call(input, **kwargs) File "D:\HGQ_DL\Nodule-CADx\src\net\nodule_net.py", line 236, in forward up1 = f_4[b, :, z_start / 4:z_end / 4, y_start / 4:y_end / 4, x_start / 4:x_end / 4].unsqueeze(0) TypeError: only integer tensors of a single element can be converted to an index

Hello , when I run this demo , I got this problem in the nodule_net.py , can you help me to solve it ? and the input is a .mhd file , it can be any file?

huangguoqing1111 commented 1 year ago

@wenyalintw very thanks for help me

huangguoqing1111 commented 1 year ago

I solve it , due to I use pytorc=1.13,so I need to change this code