torchvision/transforms/transforms.py · 11a39aaab5b55a3c116c2e8d8001bad94a96f99d · 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

transforms.py 49.2 KB

Replace transforms.py