zi2zi-pytorch icon indicating copy to clipboard operation
zi2zi-pytorch copied to clipboard

RuntimeError: output with shape [1, 256, 256] doesn't match the broadcast shape [3, 256, 256]

Open martinSCS opened this issue 1 year ago • 4 comments

train.py阶段

martinSCS avatar Apr 23 '23 19:04 martinSCS

我也碰到了,解决不了,头疼

TomTheTaller avatar May 01 '23 07:05 TomTheTaller

把train.py第45行 : "parser.add_argument('--input_nc', type=int, default=3, help='number of input images channels')" default的值改成1就行了。 这个值是dataloader识别的图片的通道数,之前我们的数据集是单通道的,这里得改成1

TomTheTaller avatar May 01 '23 08:05 TomTheTaller

input_nc的问题,几个地方默认值不一样

wftree avatar Feb 25 '24 12:02 wftree

把train.py第45行 : "parser.add_argument('--input_nc', type=int, default=3, help='number of input images channels')" default的值改成1就行了。 这个值是dataloader识别的图片的通道数,之前我们的数据集是单通道的,这里得改成1

感恩大哥啊 救我老命

not-bald-owl avatar Apr 02 '24 07:04 not-bald-owl