search

SUN-1024 / DA-TransUnet

DA-TransUNet: Combining Dual Attention of Position and Channel with Transformer U-net for Medical Image Segmentation
GNU General Public License v3.0
120 stars 10 forks source link

调试test.py时出现错误 #15

Open zfangh opened 2 months ago

zfangh commented 2 months ago

Not saving results. Traceback (most recent call last): File "test.py", line 123, in net = ViT_seg(config_vit, img_size=args.img_size, num_classes=config_vit.n_classes).cuda() File "/home/zfq/anaconda3/envs/huanjing1/DA-TransUnet-main/DA-TransUnet-main/DA-TransUNet/DA-TransUNet/Architecture/DATransUNet.py", line 407, in init self.decoder = DecoderCup(config) File "/home/zfq/anaconda3/envs/huanjing1/DA-TransUnet-main/DA-TransUnet-main/DA-TransUNet/DA-TransUNet/Architecture/DATransUNet.py", line 380, in init blocks = [ File "/home/zfq/anaconda3/envs/huanjing1/DA-TransUnet-main/DA-TransUnet-main/DA-TransUNet/DA-TransUNet/Architecture/DATransUNet.py", line 381, in DecoderBlock(in_ch, out_ch, sk_ch) for in_ch, out_ch, sk_ch in zip(in_channels, out_channels, skip_channels) File "/home/zfq/anaconda3/envs/huanjing1/DA-TransUnet-main/DA-TransUnet-main/DA-TransUNet/DA-TransUNet/Architecture/DATransUNet.py", line 325, in init self.da = DANetHead(64, 64) File "/home/zfq/anaconda3/envs/huanjing1/DA-TransUnet-main/DA-TransUnet-main/DA-TransUNet/DA-TransUNet/Architecture/block.py", line 276, in init self.sa = PAM_Module(inter_channels) File "/home/zfq/anaconda3/envs/huanjing1/DA-TransUnet-main/DA-TransUnet-main/DA-TransUNet/DA-TransUNet/Architecture/block.py", line 187, in init self.query_conv = Conv2d(in_channels=in_dim, out_channels=in_dim//8, kernel_size=1) File "/home/zfq/anaconda3/envs/huanjing1/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 330, in init super(Conv2d, self).init( File "/home/zfq/anaconda3/envs/huanjing1/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 46, in init self.reset_parameters() File "/home/zfq/anaconda3/envs/huanjing1/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 49, in reset_parameters init.kaiminguniform(self.weight, a=math.sqrt(5)) File "/home/zfq/anaconda3/envs/huanjing1/lib/python3.8/site-packages/torch/nn/init.py", line 310, in kaiminguniform fan = _calculate_correct_fan(tensor, mode) File "/home/zfq/anaconda3/envs/huanjing1/lib/python3.8/site-packages/torch/nn/init.py", line 279, in _calculate_correct_fan fan_in, fan_out = _calculate_fan_in_and_fan_out(tensor) File "/home/zfq/anaconda3/envs/huanjing1/lib/python3.8/site-packages/torch/nn/init.py", line 211, in _calculate_fan_in_and_fan_out receptive_field_size = tensor[0][0].numel() IndexError: index 0 is out of bounds for dimension 0 with size 0