nodes/eastoc-ComfyUI_SemanticSAM
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update readme

Browse Source
This commit is contained in:
eastoc
2024-08-14 03:19:26 +08:00
parent 1c3cda4e92
commit c28fa92e40
17 changed files with 331 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
node.py
+4 -4
View File
@@ -10,8 +10,8 @@ import logging
import cv2
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
sam_model_list = {"L": {"model_url": "custom_nodes/ComfyUI_Semantic_SAM/ckpt/swinl_only_sam_many2many.pth"},
"T": {"model_url": "custom_nodes/ComfyUI_Semantic_SAM/ckpt/swint_only_sam_many2many.pth"}}
sam_model_list = {"L": {"model_url": "custom_nodes/ComfyUI_SemanticSAM/ckpt/swinl_only_sam_many2many.pth"},
"T": {"model_url": "custom_nodes/ComfyUI_SemanticSAM/ckpt/swint_only_sam_many2many.pth"}}
def list_sam_model():
return list(sam_model_list.keys())
@@ -51,7 +51,7 @@ class PointPrompt():
FUNCTION = "main"
def main(self, x_coord, y_coord):
return ([[x_coord,x_coord]],)
return ([[x_coord,y_coord]],)
class SemanticSAMLoader():
@classmethod
@@ -199,4 +199,4 @@ def mask2rgba(image, mask):
#cv2.imwrite(f'mask_{i}.png', rgba_image*255)
rgba_tensor = torch.from_numpy(rgba_image)
rgba_tensor = torch.unsqueeze(rgba_tensor, dim=0)
return rgba_tensor
return rgba_tensor