test/test_transforms_tensor.py · 2cfc360e3230080b26739cb236932213a1cb03ab · OpenDAS / vision

Make RandomHorizontalFlip torchscriptable (#2282) · 11a39aaa

Francisco Massa authored Jun 04, 2020

* Make RandomHorizontalFlip torchscriptable

* Make _is_tensor_a_torch_image more generic

* Make RandomVerticalFlip torchscriptable (#2283)

* Make RandomVerticalFlip torchscriptable

* Fix lint

11a39aaa

test_transforms_tensor.py 1.4 KB

Replace test_transforms_tensor.py