wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-12 10:12:35 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Merge branch 'breifnet' of https://github.com/yousef-rafat/ComfyUI into breifnet

Browse Source
This commit is contained in:
Yousef Rafat 2026-05-08 12:04:24 +03:00
commit 640109efb3
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
comfy_extras/nodes_bg_removal.py
View File

@ -12,7 +12,7 @@ class LoadBackGroundRemovalModel(IO.ComfyNode):
node_id="LoadBackGroundRemovalModel", node_id="LoadBackGroundRemovalModel",
category="loaders", category="loaders",
inputs=[ inputs=[
IO.Combo.Input("background_removal_name", options=sorted(files)), IO.Combo.Input("bg_removal_name", options=sorted(files)),
], ],
outputs=[ outputs=[
IO.BackgroundRemoval.Output("bg_model") IO.BackgroundRemoval.Output("bg_model")
@ -30,8 +30,9 @@ class RemoveBackGround(IO.ComfyNode):
@classmethod @classmethod
def define_schema(cls): def define_schema(cls):
return IO.Schema( return IO.Schema(
node_id="RemoveBackGround", node_id="RemoveBackground",
category="encode", display_name="Remove Background",
category="image/background removal",
inputs=[ inputs=[
IO.Image.Input("image"), IO.Image.Input("image"),
IO.BackgroundRemoval.Input("bg_removal_model") IO.BackgroundRemoval.Input("bg_removal_model")
@ -45,14 +46,14 @@ class RemoveBackGround(IO.ComfyNode):
mask = bg_removal_model.encode_image(image) mask = bg_removal_model.encode_image(image)
return IO.NodeOutput(mask) return IO.NodeOutput(mask)
class BackGroundRemovalExtension(ComfyExtension): class BackgroundRemovalExtension(ComfyExtension):
@override @override
async def get_node_list(self) -> list[type[IO.ComfyNode]]: async def get_node_list(self) -> list[type[IO.ComfyNode]]:
return [ return [
LoadBackGroundRemovalModel, LoadBackgroundRemovalModel,
RemoveBackGround RemoveBackground
] ]
async def comfy_entrypoint() -> BackGroundRemovalExtension: async def comfy_entrypoint() -> BackgroundRemovalExtension:
return BackGroundRemovalExtension() return BackgroundRemovalExtension()