a hidden bug about image_size

Runinho / pytorch-cutpaste

unoffical and work in progress PyTorch implementation of CutPaste

https://runinho.github.io/pytorch-cutpaste/

229 stars 50 forks source link

a hidden bug about image_size #21

Open ShuaiLYU opened 2 years ago

ShuaiLYU commented 2 years ago

Dear Runinho,

I found that you use " h = img.size[0] w = img.size[1]" in the cutpaste.py. I think it should be " w = img.size[0] h = img.size[1]" here. the reason the bug haven't been triggered is that the images used in this dataet have same width and height.

Amending the bug is appreciated.

Thanks and Best Regards.

Runinho commented 2 years ago

Hi,

Can you create a pull request with the changes you propose?

Thanks, Runinho