上面是我的修改和添加
有两个问题:
!python /content/drive/MyDrive/pix/pix2pixHD-master/train.py --label_nc 0 --ngf 64 --no_instance --dataroot /content/drive/MyDrive/pix/pix2pixHD-master/datasets/cityscapes/
这样运行,报错是:RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 819 but got size 832 for tensor number 1 in the list.
这是大小不一样吗?可是我是在同一个数据集里面摘出来的
1、输入数据是需要预处理成同样维度的,第一维度一般是batch,可以不同,其他维度表示图片的通道、高度、宽度等信息,需要相同。
2、你这个模型的作用是:输入黑白图,输出彩色图,在test测试集中,test_A作为输入,然后模型输出图片,用来和test_B里面的图片进行验证。
应该是维度的处理问题吧,可以查一下;
是想输出彩色图和test_B验证,这样没毛病吧