diff --git a/impact_pack.py b/impact_pack.py index 2c0296e..04cd893 100644 --- a/impact_pack.py +++ b/impact_pack.py @@ -1133,7 +1133,7 @@ class SegmDetectorForEach: CATEGORY = "ImpactPack/Detector" def doit(self, segm_model, image, threshold, dilation, crop_factor): - mmdet_results = inference_segm(segm_model, image, threshold) + mmdet_results = inference_segm(image, segm_model, threshold) segmasks = create_segmasks(mmdet_results) if dilation > 0: @@ -1391,7 +1391,7 @@ class SegmDetectorCombined: CATEGORY = "ImpactPack/Detector" def doit(self, segm_model, image, threshold, dilation): - mmdet_results = inference_segm_old(segm_model, image, threshold) + mmdet_results = inference_segm(image, segm_model, threshold) segmasks = create_segmasks(mmdet_results) if dilation > 0: segmasks = dilate_masks(segmasks, dilation)