torchvision/ops/ps_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

ps_roi_align.py 2.93 KB

Replace ps_roi_align.py