MaybeShewill-CV / attentive-gan-derainnet

Unofficial tensorflow implemention of "Attentive Generative Adversarial Network for Raindrop Removal from A Single Image (CVPR 2018) " model https://maybeshewill-cv.github.io/attentive-gan-derainnet/
MIT License
262 stars 83 forks source link

训练自己图片集有问题 #72

Closed kaixinwangzi1982 closed 4 years ago

kaixinwangzi1982 commented 4 years ago

你好,我用master最新代码训练自己的图片集,图片后缀名是jpg格式的,头两万次训练正常,G_loss越来越低,用2000次的checkpoint测试的时候,图片正常。 随着训练次数曾多,G_loss偶尔会出现负数,后来负数越来越多,用后面的checkpoint derain图片的时候,图片是全黑色的。不知道是哪里出问题了。 同样的图片训练集在release代码里没有这种情况