nodes/ltdrdata-ComfyUI-Impact-Pack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

segmdetector patch again...

Browse Source
This commit is contained in:
Dr.Lt.Data
2023-05-01 21:10:43 +09:00
parent 938ef008a8
commit abb4e864e1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
impact_pack.py
+2 -2
View File
@@ -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)