Added Support for PIL images for CutOut in transform/tranforms.py

PistonY / torch-toolbox

🛠 Toolbox to extend PyTorch functionalities

BSD 3-Clause "New" or "Revised" License

417 stars 56 forks source link

Added Support for PIL images for CutOut in transform/tranforms.py #7

Closed ArthDh closed 4 years ago

ArthDh commented 4 years ago

Possible fix to #6

PistonY commented 4 years ago

Hi, thanks for working on this pr. Please let me know if you ready to merge this(finish work). Before this please do some sample tests on your local device.

ArthDh commented 4 years ago

Hey @PistonY! Tested on JPEG, PNG and GIF files loaded through PIL, looks good to me now! Do let me know if anything is to be changed. Thanks!