search

google-research / google-research

Google Research
https://research.google
Apache License 2.0
34.08k stars 7.86k forks source link

flare_removal: Error - Different batch tensors #1447

Open SevFle opened 1 year ago

SevFle commented 1 year ago

Hi,

I'm trying to train the network, listed in flare_removal directory. Following the provided tutorial, I always get an error about different tensor shape when reaching line 132 of train.py.

Cannot batch tensors with different shapes in component 0. First element had shape [738,1126,3] and element 1 had shape [770,1076,3]. [Op:IteratorGetNext]

Do I have to reshape all Input Images first to fit the provided lensflare image shape?

Best regards, Severin

charles92 commented 1 year ago

Hi Severin,

Thanks for your interest in our work and apologies for the late reply. We have not encountered issues like this before, and the symptom you described makes me suspect image size mismatch, too.

Would it work if you resize/crop all your inputs to 512x512 before running the code? I don't remember exactly whether it was necessary (it's been a while and I've already left the team), but it's probably worth trying?

Thanks, Qiurui