CavinHuang-comfyui-nodes-docs/docs/LayerUtility%3A GetImageSize.md

Documentation

• Class name: GetImageSize
• Category: 😺dzNodes/LayerUtility
• Output node: False
• Repo Ref: https://github.com/chflame163/ComfyUI_LayerStyle

获取图片的宽度和高度。

Input types

Required

• image
  • 输入的图像。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Output types

• width

  • 图片的宽度。
  • Comfy dtype: INT
  • Python dtype: int

• height

  • 图片的高度。
  • Comfy dtype: INT
  • Python dtype: int

• original_size

  • 图片的原始尺寸。
  • Comfy dtype: BOX
  • Python dtype: list

Usage tips

• Infra type: CPU

Source code

class GetImageSize:

    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(self):

        return {
            "required": {
                "image": ("IMAGE", ),
            },
            "optional": {
            }
        }

    RETURN_TYPES = ("INT", "INT", "BOX")
    RETURN_NAMES = ("width", "height",  "original_size")
    FUNCTION = 'get_image_size'
    CATEGORY = '😺dzNodes/LayerUtility'

    def get_image_size(self, image,):

        if image.shape[0] > 0:
            image = torch.unsqueeze(image[0], 0)
        _image = tensor2pil(image)

        return (_image.width, _image.height, [_image.width, _image.height],)