Relative import for roi_align (#4533)

pytorch · Oct 4, 2021 · e45489b · e45489b
1 parent 05d7651
commit e45489b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/ops/poolers.py b/torchvision/ops/poolers.py
@@ -2,7 +2,7 @@
 from torch import nn, Tensor
 
 import torchvision
-from torchvision.ops import roi_align
+from .roi_align import roi_align
 from torchvision.ops.boxes import box_area
 
 from typing import Optional, List, Dict, Tuple, Union