关于其他数据集的问题

hzwer / ICCV2019-LearningToPaint

ICCV2019 - Learning to Paint With Model-based Deep Reinforcement Learning

MIT License

2.25k stars 312 forks source link

Closed Donkdot closed 1 year ago

Donkdot commented 1 year ago

您好！我在使用CUB Birds 和 Stanford cars数据集进行训练时，图片只显示一个颜色，随着训练过程进行也没有其他变化，我对代码的修改仅有load_data(), 为什么会造成这种情况呢？

hzwer commented 1 year ago

你好，按理说一开始模型的输出如果是随机的，那颜色也应该是各种都有。是不是有可能你的输入图片范围变成了 0-255 ？如果说只换了数据集，我暂时想不到其它的可能问题。

Donkdot commented 1 year ago

好的，我再检查一下，感谢您的回复！