Giters

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/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

the bug

crazygirl1992 opened this issue · comments

commented

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]]

commented

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

commented

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

commented

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