torchvision/ops/roi_align.py · 971c3e45b96bc5aa5868c45cd40e4f3c3d90d126 · OpenDAS / vision

Type annotations for torchvision.ops (#2331) · 971c3e45

Vukašin Manojlović authored Jul 06, 2020

* Add type annotations for torchvision.ops

* Fix type annotations for torchvision.ops

* Fix typo in import

* Fix undefined name in FeaturePyramidNetwork

971c3e45

roi_align.py 3.18 KB

Replace roi_align.py