python实现mixup数据增强

最近yolov4也出来了,不说其他的tricks,但是数据增强中,mixup这个方法就在最近的文章中多次被提到,简单的说就是两张图片合成一张。在此我们使用PIL来实现方法,直接上代码:

from PIL import Image
img1 = Image.open("object.jpg")
img1 = img1.convert('RGBA')

img2 = Image.open("other.jpg")
img2 = img2.convert('RGBA')
#resize to size of img1
img2 = img2.resize(img1.size)
print(img2.size)

img = Image.blend(img1, img2, 0.2)
img.show()
img.save("blend.jpg")

效果:

python实现mixup数据增强_第1张图片

python实现mixup数据增强_第2张图片

自己在代码中需要按照自己的需求来设置一些resize的规则。

你可能感兴趣的:(Python,算法,其他)