2020-09-10
问题:RuntimeError: Given groups=1, weight of size 16 1 3 3, expected input[2, 3, 128, 128] to have 1 channels, but got 3 channels instead是因为pytorch读入图片默认为RGB三通道(即使你是灰度图它也是读取灰度图),我将in_channels=1,读取灰度图依然有错,改正方式:在train_transform = transforms.Compose()中加入trans
原创
2020-09-10 17:40:34 ·
360 阅读 ·
0 评论