Merge pull request #219 from vineet131/master

Fix center_size import
WisconsinAIVision · Nov 4, 2022 · 76cf868 · 76cf868
2 parents 7d692ff + 3a359a8
commit 76cf868
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolact_edge/layers/output_utils.py b/yolact_edge/layers/output_utils.py
@@ -10,7 +10,7 @@
 from yolact_edge.data import cfg, mask_type, MEANS, STD, activation_func
 from yolact_edge.utils.augmentations import Resize
 from yolact_edge.utils import timer
-from .box_utils import crop, sanitize_coordinates
+from .box_utils import crop, sanitize_coordinates, center_size
 
 def postprocess(det_output, w, h, batch_idx=0, interpolation_mode='bilinear',
                 visualize_lincomb=False, crop_masks=True, score_threshold=0):