wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-10 17:22:36 +08:00
Code Issues Actions 13 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",
category="loaders",
inputs=[
IO.Combo.Input("background_removal_name", options=sorted(files)),
IO.Combo.Input("bg_removal_name", options=sorted(files)),
],
outputs=[
IO.BackgroundRemoval.Output("bg_model")
@ -30,8 +30,9 @@ class RemoveBackGround(IO.ComfyNode):
@classmethod
def define_schema(cls):
return IO.Schema(
node_id="RemoveBackGround",
category="encode",
node_id="RemoveBackground",
display_name="Remove Background",
category="image/background removal",
inputs=[
IO.Image.Input("image"),
IO.BackgroundRemoval.Input("bg_removal_model")
@ -45,14 +46,14 @@ class RemoveBackGround(IO.ComfyNode):
mask = bg_removal_model.encode_image(image)
return IO.NodeOutput(mask)
class BackGroundRemovalExtension(ComfyExtension):
class BackgroundRemovalExtension(ComfyExtension):
@override
async def get_node_list(self) -> list[type[IO.ComfyNode]]:
return [
LoadBackGroundRemovalModel,
RemoveBackGround
LoadBackgroundRemovalModel,
RemoveBackground
]
async def comfy_entrypoint() -> BackGroundRemovalExtension:
return BackGroundRemovalExtension()
async def comfy_entrypoint() -> BackgroundRemovalExtension:
return BackgroundRemovalExtension()