From 872d449bdbb09a1c370567aacc9aaba866352a04 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 4 Oct 2021 11:23:57 +0100 Subject: [PATCH] Relative import for roi_align --- torchvision/ops/poolers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/ops/poolers.py b/torchvision/ops/poolers.py index f4ff289299b..8438687c093 100644 --- 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