SimCLR，增强生成一张图像而不是两张

Posted 2023-02-23

【中文标题】SimCLR，增强生成一张图像而不是两张

【英文标题】：SimCLR, augmentation generates one images instead of two

【发布时间】：2021-09-11 16:57:35

【问题描述】：

我正在研究 SimCLR 的增强，我们必须从一张图像生成两个。但如果我绘制我的输出，它只显示一个图像。我不明白为什么

def augment_ops()
....
    return image
 def preprocess_image_train(image):
   
    image1 = augment_ops(image)
    image2 = augment_ops(image)

    return image1, image2
output = preprocess_image_train(img1)
plt.imshow(output)
plt.show()

【参考方案1】：

你不应该调用这个函数吗，因为返回了两个图像，像这样：

output1,output2 = preprocess_image_train(img1)

然后分别显示？

plt.imshow(output1)
plt.show()

plt.imshow(output2)
plt.show()