search

bigmb / Unet-Segmentation-Pytorch-Nest-of-Unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet
MIT License
1.87k stars 345 forks source link

tuple index out of range #50

Closed pangda72 closed 3 years ago

pangda72 commented 3 years ago

hi,bigmb when i use AttU_Net,i got a problem:

===================================================

Traceback (most recent call last): File "C:/Users/lenovo/Desktop/pspnet/unet1.py", line 382, in summary(model,(3,256,256)) File "E:\Anaconda3\envs\pytorch\lib\site-packages\torchsummary\torchsummary.py", line 72, in summary model(x) File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(input, **kwargs) File "C:/Users/lenovo/Desktop/pspnet/unet1.py", line 349, in forward x4 = self.Att5(g=d5, x=e4) File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 549, in call hook_result = hook(self, input, result) File "E:\Anaconda3\envs\pytorch\lib\site-packages\torchsummary\torchsummary.py", line 19, in hook summary[m_key]["input_shape"] = list(input[0].size()) IndexError: tuple index out of range

===============================================================

I don't know how to solve it ,could you help me .Thank you very much! Looking forward to your reply. Best.

bigmb commented 3 years ago

Hello @pangda72 ,

Please look into this issue. [https://github.com/bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets/issues/12]

pangda72 commented 3 years ago

Hello @pangda72 ,

Please look into this issue. [https://github.com/[/issues/12](https://github.com/bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets/issues/12)]

Okay,bigmb.Thank you for your prompt reply,I'm going to look into that link.