search

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

the bug #76

Closed crazygirl1992 closed 4 years ago

crazygirl1992 commented 4 years ago

hellow,i can run the test code,but,when i run the train code,this bug happened, why this happened?

tensorflow.python.framework.errors_impl.InvalidArgumentError: Input to reshape is a tensor with 288768 values, but the requested shape has 786432 [[val_IteratorGetNext]]

MaybeShewill-CV commented 4 years ago

@crazygirl1992 You must have used images with wrong image shape. Generate tfrecords first and then do the training procedure

crazygirl1992 commented 4 years ago

thank you, i have resolved this problem,and why i run the code,it use cpu nor GPU,i have the same arg setting

MaybeShewill-CV commented 4 years ago

@crazygirl1992 You must installed tensorflow gpu incorrectly. That's a environment relative problem and there should be nothing wrong with this code repo:)