Update AILab_SegmentV2.py

1038lab · web-flow · commit a3bc7460b370 · 2025-06-01T14:25:45.000-07:00
diff --git a/AILab_SegmentV2.py b/AILab_SegmentV2.py
@@ -268,10 +268,10 @@ def segment_v2(self, image, prompt, sam_model, dino_model, threshold=0.30,
         return (pil2tensor(result_image), mask_tensor, mask_image_vis)
 
 NODE_CLASS_MAPPINGS = {
-    "AILab_SegmentV2": SegmentV2,
+    "SegmentV2": SegmentV2,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "AILab_SegmentV2": "Segmentation V2 (RMBG)",
+    "SegmentV2": "Segmentation V2 (RMBG)",
 }
 

Original file line number	Diff line number	Diff line change
`@@ -268,10 +268,10 @@ def segment_v2(self, image, prompt, sam_model, dino_model, threshold=0.30,`
`268`	`268`	`return (pil2tensor(result_image), mask_tensor, mask_image_vis)`
`269`	`269`
`270`	`270`	`NODE_CLASS_MAPPINGS = {`
`271`		`- "AILab_SegmentV2": SegmentV2,`
	`271`	`+ "SegmentV2": SegmentV2,`
`272`	`272`	`}`
`273`	`273`
`274`	`274`	`NODE_DISPLAY_NAME_MAPPINGS = {`
`275`		`- "AILab_SegmentV2": "Segmentation V2 (RMBG)",`
	`275`	`+ "SegmentV2": "Segmentation V2 (RMBG)",`
`276`	`276`	`}`
`277`	`277`