bala1802/StableDiffusionModel

1import torch2import torchvision.transforms as T3import torch.nn.functional as F4 5def blue_channel(images):6    error = torch.abs(images[:,2] - 0.9).mean() 7    return error8 9def elastic_transform(images):10    elastic_transformer = T.ElasticTransform(alpha=550.0,sigma=5.0)11    transformed_imgs = elastic_transformer(images)12    error = torch.abs(transformed_imgs - images).mean()13    return error14 15def symmetry(images):16    flipped_image = torch.flip(images, [3])17    error = F.mse_loss(images, flipped_image)18    print("Loss Calculated for the Symmetry : ", error)19    return error20 21def saturation(images):22    transformed_imgs = T.functional.adjust_saturation(images,saturation_factor = 10)23    error = torch.abs(transformed_imgs - images).mean()24    return error