Solved tensor size mismatch error in test own images

IsHYuhi / ST-CGAN_Stacked_Conditional_Generative_Adversarial_Networks

Unofficial implementation of ''Stacked Conditional Generative Adversarial Networks for Jointly Learning Shadow Detection and Shadow Removal'' with PyTorch

67 stars 15 forks source link

Solved tensor size mismatch error in test own images #10

Closed Param-Raval closed 4 years ago

Param-Raval commented 4 years ago

While testing on own images the test_own_image function returned a tensor size mismatch error. Solved it by adding a couple of parameters in utils/data_loader.py (line 54).

IsHYuhi commented 4 years ago

Thank you for your work. However, using RandomCrop is not preferable because your images would be cropped. The cause of size mismatch is that the input image was resized. I modified the code in #11 . If any other problems, please open an issue.

Thank you.