problem in train_baseline.py
RobinWenqian opened this issue · comments
感觉不应该啊,如果你实在要改的话,直接在train_baseline.py里面的 dcganDataset这个类里面改,这个类是我自己定义的一个图片读取的类, 你可以根据这个修改, 你把这个逻辑看懂后, 对于pytorch这个框架,几乎任何一种数据读取方式就都会了
好的好的!十分感谢!
检查下你的数据是不是有问题,还有文件夹的名字
文件夹什么的就是下载下来数据集之后用prepare.py和changeIndex.py处理过后的
temp里面folder加上下划线再加上filename是我为了解析出文件的名字及其对应存储的位置,你不用下划线用其他符号都行,文件夹的名字表示文件夹中图片的label, 因为生成的图片和原始的训练图片的损失函数不同,所以我用了一个flag来进行区分判断,
好的,十分感谢!我去改改试试
彻底将读文件的方式更改了,用了pytorch标准库的ImageFolder类。正在跑结果,谢谢帮助
彻底将读文件的方式更改了,用了pytorch标准库的ImageFolder类。正在跑结果,谢谢帮助
同学你好,我想问一下你读文件的方式是如何更改的,我现在读生成的图像出现了下面的问题,我已经更改过了但是还是不行,希望同学可以帮我,非常感谢!
https://github.com/qiaoguan/Person-reid-GAN-pytorch/issues/26