mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-30 00:00:26 +08:00
Merge c7843f888f into 034fac7054
This commit is contained in:
commit
82a4ba6a57
@ -1128,6 +1128,25 @@ class ImageCompare(ComfyTypeI):
|
|||||||
def as_dict(self):
|
def as_dict(self):
|
||||||
return super().as_dict()
|
return super().as_dict()
|
||||||
|
|
||||||
|
@comfytype(io_type="BOUNDINGBOX")
|
||||||
|
class BoundingBox(ComfyTypeIO):
|
||||||
|
Type = dict
|
||||||
|
|
||||||
|
class Input(WidgetInput):
|
||||||
|
def __init__(self, id: str, display_name: str=None, optional=False, tooltip: str=None,
|
||||||
|
socketless: bool=True, default: dict=None, component: str=None):
|
||||||
|
super().__init__(id, display_name, optional, tooltip, None, default, socketless)
|
||||||
|
self.component = component
|
||||||
|
if default is None:
|
||||||
|
self.default = {"x": 0, "y": 0, "width": 512, "height": 512}
|
||||||
|
|
||||||
|
def as_dict(self):
|
||||||
|
d = super().as_dict()
|
||||||
|
if self.component:
|
||||||
|
d["component"] = self.component
|
||||||
|
return d
|
||||||
|
|
||||||
|
|
||||||
DYNAMIC_INPUT_LOOKUP: dict[str, Callable[[dict[str, Any], dict[str, Any], tuple[str, dict[str, Any]], str, list[str] | None], None]] = {}
|
DYNAMIC_INPUT_LOOKUP: dict[str, Callable[[dict[str, Any], dict[str, Any], tuple[str, dict[str, Any]], str, list[str] | None], None]] = {}
|
||||||
def register_dynamic_input_func(io_type: str, func: Callable[[dict[str, Any], dict[str, Any], tuple[str, dict[str, Any]], str, list[str] | None], None]):
|
def register_dynamic_input_func(io_type: str, func: Callable[[dict[str, Any], dict[str, Any], tuple[str, dict[str, Any]], str, list[str] | None], None]):
|
||||||
DYNAMIC_INPUT_LOOKUP[io_type] = func
|
DYNAMIC_INPUT_LOOKUP[io_type] = func
|
||||||
@ -2049,4 +2068,5 @@ __all__ = [
|
|||||||
"ImageCompare",
|
"ImageCompare",
|
||||||
"PriceBadgeDepends",
|
"PriceBadgeDepends",
|
||||||
"PriceBadge",
|
"PriceBadge",
|
||||||
|
"BoundingBox",
|
||||||
]
|
]
|
||||||
|
|||||||
@ -26,16 +26,18 @@ class ImageCrop(IO.ComfyNode):
|
|||||||
category="image/transform",
|
category="image/transform",
|
||||||
inputs=[
|
inputs=[
|
||||||
IO.Image.Input("image"),
|
IO.Image.Input("image"),
|
||||||
IO.Int.Input("width", default=512, min=1, max=nodes.MAX_RESOLUTION, step=1),
|
IO.BoundingBox.Input("crop_region", component="ImageCrop"),
|
||||||
IO.Int.Input("height", default=512, min=1, max=nodes.MAX_RESOLUTION, step=1),
|
|
||||||
IO.Int.Input("x", default=0, min=0, max=nodes.MAX_RESOLUTION, step=1),
|
|
||||||
IO.Int.Input("y", default=0, min=0, max=nodes.MAX_RESOLUTION, step=1),
|
|
||||||
],
|
],
|
||||||
outputs=[IO.Image.Output()],
|
outputs=[IO.Image.Output()],
|
||||||
)
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def execute(cls, image, width, height, x, y) -> IO.NodeOutput:
|
def execute(cls, image, crop_region) -> IO.NodeOutput:
|
||||||
|
x = crop_region.get("x", 0)
|
||||||
|
y = crop_region.get("y", 0)
|
||||||
|
width = crop_region.get("width", 512)
|
||||||
|
height = crop_region.get("height", 512)
|
||||||
|
|
||||||
x = min(x, image.shape[2] - 1)
|
x = min(x, image.shape[2] - 1)
|
||||||
y = min(y, image.shape[1] - 1)
|
y = min(y, image.shape[1] - 1)
|
||||||
to_x = width + x
|
to_x = width + x
|
||||||
@ -46,6 +48,27 @@ class ImageCrop(IO.ComfyNode):
|
|||||||
crop = execute # TODO: remove
|
crop = execute # TODO: remove
|
||||||
|
|
||||||
|
|
||||||
|
class IntToBoundingBox(IO.ComfyNode):
|
||||||
|
@classmethod
|
||||||
|
def define_schema(cls):
|
||||||
|
return IO.Schema(
|
||||||
|
node_id="IntToBoundingBox",
|
||||||
|
display_name="INT to Bounding Box",
|
||||||
|
category="utils",
|
||||||
|
inputs=[
|
||||||
|
IO.Int.Input("x", default=0, min=0, max=MAX_RESOLUTION),
|
||||||
|
IO.Int.Input("y", default=0, min=0, max=MAX_RESOLUTION),
|
||||||
|
IO.Int.Input("width", default=512, min=1, max=MAX_RESOLUTION),
|
||||||
|
IO.Int.Input("height", default=512, min=1, max=MAX_RESOLUTION),
|
||||||
|
],
|
||||||
|
outputs=[IO.BoundingBox.Output(display_name="BOUNDINGBOX")],
|
||||||
|
)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def execute(cls, x, y, width, height) -> IO.NodeOutput:
|
||||||
|
return IO.NodeOutput({"x": x, "y": y, "width": width, "height": height})
|
||||||
|
|
||||||
|
|
||||||
class RepeatImageBatch(IO.ComfyNode):
|
class RepeatImageBatch(IO.ComfyNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def define_schema(cls):
|
def define_schema(cls):
|
||||||
@ -628,6 +651,7 @@ class ImagesExtension(ComfyExtension):
|
|||||||
async def get_node_list(self) -> list[type[IO.ComfyNode]]:
|
async def get_node_list(self) -> list[type[IO.ComfyNode]]:
|
||||||
return [
|
return [
|
||||||
ImageCrop,
|
ImageCrop,
|
||||||
|
IntToBoundingBox,
|
||||||
RepeatImageBatch,
|
RepeatImageBatch,
|
||||||
ImageFromBatch,
|
ImageFromBatch,
|
||||||
ImageAddNoise,
|
ImageAddNoise,
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user