用Python实现 EmptyLatentImage类

类 EmptyLatentImage

功能

生成一个指定大小的空潜在空间图像（所有像素值为零）。

构造函数

• __init__：初始化类实例，设置计算设备为中间设备。

类方法 INPUT_TYPES

• 定义输入参数的类型和默认值，用于生成空潜在空间图像。

方法 generate

• 生成一个指定宽度、高度和批次大小的空潜在空间图像。

详细说明

class EmptyLatentImage:
    def __init__(self):
        self.device = comfy.model_management.intermediate_device()

初始化

• self.device 设置为中间设备，通过 comfy.model_management.intermediate_device() 获取。

    @classmethod
    def INPUT_TYPES(s):
        return {"required": { "width": ("INT", {"default": 512, "min": 16, "max": MAX_RESOLUTION, "step": 8}),
                              "height": ("INT", {"default": 512, "min": 16, "max": MAX_RESOLUTION, "step": 8}),
                              "batch_size": ("INT", {"default": 1, "min": 1, "max": 4096})}}

定义输入参数

• width：图像宽度，默认值为512，最小值为16，最大值为 MAX_RESOLUTION，步长为8。
• height：图像高度，默认值为512，最小值为16，最大值为 MAX_RESOLUTION，步长为8。
• batch_size：批次大小，默认值为1，最小值为1，最大值为4096。

    RETURN_TYPES = ("LATENT",)
    FUNCTION = "generate"

    CATEGORY = "latent"

返回类型和分类

• RETURN_TYPES：返回类型为 "LATENT"。
• FUNCTION：指定方法为 generate。
• CATEGORY：分类为 "latent"。

    def generate(self, width, height, batch_size=1):
        latent = torch.zeros([batch_size, 4, height // 8, width // 8], device=self.device)
        return ({"samples": latent}, )

生成方法

• generate 方法用于生成空潜在空间图像。
• 使用 torch.zeros 创建一个全零张量，其形状为 [batch_size, 4, height // 8, width // 8]。
  • batch_size：批次大小。
  • 4：通常表示潜在空间的通道数。
  • height // 8 和 width // 8：图像的高度和宽度除以8，这是因为潜在空间图像的分辨率通常比原始图像低（如原始图像大小的一半或更小）。
• 返回值是一个包含生成的潜在空间图像的字典。

总结

这个类的主要作用是生成空的潜在空间图像，用于在图像生成和处理任务中作为输入。它定义了必要的输入参数，生成方法，并返回潜在空间图像的字典形式。