注釈付きの例

i = Image.open(image_path)
i = ImageOps.exif_transpose(i)
if i.mode == 'I':
    i = i.point(lambda i: i * (1 / 255))
image = i.convert("RGB")
image = np.array(image).astype(np.float32) / 255.0
image = torch.from_numpy(image)[None,]

for (batch_number, image) in enumerate(images):
    i = 255. * image.cpu().numpy()
    img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
    filepath = # バッチ番号を考慮したパス
    img.save(filepath)

mask = 1.0 - mask

# [B,H,W,C] が必要です。ここで C = 1
if len(mask.shape)==2: # [H,W] のため、B と C を次元 1 として挿入
    mask = mask[None,:,:,None]
elif len(mask.shape)==3 and mask.shape[2]==1: # [H,W,C] の場合
    mask = mask[None,:,:,:]
elif len(mask.shape)==3:                      # [B,H,W] の場合
    mask = mask[:,:,:,None]

# マスクを元の透過層に戻すために反転
mask = 1.0 - mask

# `C`（チャンネル）次元を Unsqueeze
mask = mask.unsqueeze(-1)

# `C` 次元に沿って結合（"cat"）
rgba_image = torch.cat((rgb_image, mask), dim=-1)

class Noise_MixedNoise:
    def __init__(self, nosie1, noise2, weight2):
        self.noise1  = noise1
        self.noise2  = noise2
        self.weight2 = weight2

    @property
    def seed(self): return self.noise1.seed

    def generate_noise(self, input_latent:torch.Tensor) -> torch.Tensor:
        noise1 = self.noise1.generate_noise(input_latent)
        noise2 = self.noise2.generate_noise(input_latent)
        return noise1 * (1.0-self.weight2) + noise2 * (self.weight2)