m0_62267751 2022-07-19 16:54 采纳率: 85.7%
浏览 206
已结题

用自己的数据集测试pix2pixHD,下面是我的一些操作,有两个问题

img

img

img

img

上面是我的修改和添加
有两个问题:


!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. 我在运行时,流程是先进行train ,之后再进行test ,在test_B可以生成test_A里面的图片吗
  • 写回答

2条回答 默认 最新

  • 脚踏南山 2022-07-19 17:28
    关注

    1、输入数据是需要预处理成同样维度的,第一维度一般是batch,可以不同,其他维度表示图片的通道、高度、宽度等信息,需要相同。
    2、你这个模型的作用是:输入黑白图,输出彩色图,在test测试集中,test_A作为输入,然后模型输出图片,用来和test_B里面的图片进行验证。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 7月30日
  • 已采纳回答 7月22日
  • 修改了问题 7月19日
  • 修改了问题 7月19日
  • 展开全部