Update coco_instance_new_baseline_dataset_mapper.py

AstroBoy1 · Dec 9, 2022 · 78f3c26 · 78f3c26
1 parent 8a9fbc2
commit 78f3c26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/detrex/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py b/detrex/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py
@@ -25,7 +25,7 @@
 import numpy as np
 import torch
 
-from detectron2.structures import PolygonMasks
+from detectron2.structures import Instances, Boxes, PolygonMasks
 from detectron2.data import detection_utils as utils
 from detectron2.data import transforms as T