Unverified Commit e45489b1 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Relative import for roi_align (#4533)

parent 05d76519
......@@ -2,7 +2,7 @@ import torch
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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment