Demo cannot handle input images of non 512x512 size

Picsart-AI-Research / MI-GAN

[ICCV 2023] MI-GAN: A Simple Baseline for Image Inpainting on Mobile Devices

MIT License

455 stars 44 forks source link

Demo cannot handle input images of non 512x512 size #17

Open MarcusW03 opened 1 week ago

MarcusW03 commented 1 week ago

I was running the demo of the mi-gan with my own set of testing images. I noticed that for images that aren't 512x512 an error is produced. `/scripts/demo.py", line 140, in main composed_img = img_resized mask_resized + result_image (1 - mask_resized)


ValueError: operands could not be broadcast together with shapes (341,512,3) (340,512,1) `

MarcusW03 commented 1 week ago

The image that produced the above error has size of 1400x933

MarcusW03 commented 4 days ago

Submitted Pull Request https://github.com/Picsart-AI-Research/MI-GAN/pull/18 for this issue.