数据增强之mixup

最新推荐文章于 2024-06-09 23:04:45 发布

WhaleYu2003

最新推荐文章于 2024-06-09 23:04:45 发布

阅读量718

点赞数

分类专栏：深度学习小技巧文章标签： opencv 计算机视觉 python

本文链接：https://blog.csdn.net/weixin_59605625/article/details/127573010

版权

深度学习小技巧专栏收录该内容

1 篇文章 0 订阅

订阅专栏

数据增强之mixup，就是做一个图像融合，简单理解就是两张图像的叠加。具体细节见：https://blog.csdn.net/u013685264/article/details/122622919

具体效果如下：
原图：
在这里插入图片描述

mixup后：
在这里插入图片描述

代码实现：

import cv2
import numpy as np


# MixUP
img1 = cv2.imread('dog.png')
img1 = cv2.resize(img1, (1960, 1080))

img2 = cv2.imread('cat.png')
img2 = cv2.resize(img2, (1960, 1080))

alpha = 1.0
lam = np.random.beta(alpha, alpha)
print(lam)

mixed_img = lam * img1 + (1 - lam) * img2

cv2.imwrite('mixed_img.jpeg', mixed_img)