hassony2 / torch_videovision

Transforms for video datasets in pytorch
GNU General Public License v3.0
268 stars 44 forks source link

Bug in RandomResizedCrop #18

Open Ghelman-A opened 2 years ago

Ghelman-A commented 2 years ago

Hi @hassony2

The last line in the RandomResizedCrop should be changed from

https://github.com/hassony2/torch_videovision/blob/785a810d1df2ccab6901fb98b38e358433e7b17a/torchvideotransforms/video_transforms.py#L312

to return F.resize_clip(imgs,self.size,self.interpolation) for the cropping to be applied as well.