【踩雷小记】pytorch用transforms同时旋转图像和标签

最新推荐文章于 2024-10-12 12:26:23 发布

翻滚的年糕

最新推荐文章于 2024-10-12 12:26:23 发布

阅读量1.9k

点赞数 2

文章标签： 1024程序员节 python pytorch

本文链接：https://blog.csdn.net/weixin_45760688/article/details/127487755

版权

这篇博客探讨了在图像处理中如何使用概率参数进行随机变换，如水平和垂直翻转、灰度化，并确保这些变换同时应用于图像和对应的标签。通过random.randint()函数确定0或1的概率值，保证每次处理时图像和标签的操作一致。举例说明了如何使用transforms.Compose()组合多个变换，并应用到图像和标签上。

摘要由CSDN通过智能技术生成

对于transforms中带有概率参数的函数，例如：

transforms.RandomHorizontalFlip(p=0.5) 依概率p进行水平翻转

transforms.RandomVerticalFlip(p=0.5) 依概率p进行垂直翻转

transforms.RandomGrayscale(p=0.1) 依概率p转化为灰度图

为了使图像和标签能够同时执行相同操作，需要先将概率p确定为0或1，但这个确定的过程可以是随机的结果：

p = random.randint(0,1) #返回 [0，1] 之间的任意整数

然后基于确定的结果p对图像和标签同时使用transforms函数即可。

每张图像可以重新确定p值。

如：

p = random.randint(0,1)

transform = transforms.Compose([ transforms.RandomHorizontalFlip(p), transforms.RandomVerticalFlip(p)

])

img = transform(image)

label = transform(label)