if img_ch =1 ，output with shape doesn't match the broadcast shape

LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.

2.68k stars 595 forks source link

if img_ch =1 ，output with shape doesn't match the broadcast shape #37

Closed mk123qwe closed 1 year ago

mk123qwe commented 5 years ago

if img_ch =1 ，output with shape doesn't match the broadcast shape Input is grayscale image. Thanks!

mk123qwe commented 5 years ago

if img_ch = 3, Input is grayscale image transfer to channel 3 another issue ValueError: Target and input must have the same number of elements. target nelement (196608) != input nelement (65536)