用于目标检测和图像分类任务的图像增强库。
augment-auto的Python项目详细描述
增强自动
基于opencv和numpy的python图像增强库。它可以用于图像分类和目标检测任务中的图像增强。支持许多不同的扩充技术,这些技术可分为三种主要类型:几何变换、光度变换和基于核的变换。 库也支持带边框的图像。在
安装
使用pip安装:
pipinstallaugment-auto
从github安装:
文档
完整文档-https://augment-auto.readthedocs.io/en/latest/
演示ipython记事本-demo.ipynb
示例
# Geometric Transformationsimg=cv2.imread('images/3.jpg')img_new=crop(img,point1=(100,100),point2=(450,400))img_new=rotate(img,angle=15,keep_resolution=True)img_new=scale(img,fx=1.5,fy=1.5,keep_resolution=False)img_new=shear(img,shear_val=0.2,axis=1)img_new=translate(img,tx=50,ty=60)
# Geometric Transformations with bounding boximg=cv2.imread('images/0.jpeg')bbox=[581,274,699,321]img_new,bbox_new=crop(img,point1=(100,100),point2=(650,400),box=bbox)img_new,bbox_new=rotate(img,angle=15,keep_resolution=True,box=bbox)img_new,bbox_new=scale(img,fx=1.5,fy=1.3,keep_resolution=False,box=bbox)img_new,bbox_new=shear(img,shear_val=0.2,axis=0,box=bbox)img_new,bbox_new=translate(img,tx=50,ty=160,box=bbox)
# Photometric Transformationsimg=cv2.imread('images/1.jpg')img=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)img_new=brightness_contrast(img,alpha=1.3,beta=20)img_new=brightness_contrast(img,alpha=0.7,beta=-10)img_new=colorSpace(img,colorspace='hsv')img_new=colorSpace(img,colorspace='ycrcb')img_new=colorSpace(img,colorspace='lab')img_new=addNoise(img,'gaussian',mean=0,var=0.08)img_new=addNoise(img,'salt_pepper',sp_ratio=0.5,noise_amount=0.1)img_new=addNoise(img,'poisson',noise_amount=0.5)
# Kernel-based Transformationsimg=cv2.imread('images/0.jpeg')bbox=[581,274,699,321]img_new=randomErase(img,size=(100,100))img_new=randomCropAdd(img,size=(100,100))img_new=sharpen(img)img_new=randomErase(img,size=(60,40),box=bbox)img_new=randomCropAdd(img,size=(60,40),box=bbox)img_new=blur(img,'avg',ksize=(9,9))img_new=blur(img,'gaussian',ksize=(9,9),gaussian_sigma=0)img_new=blur(img,'median',median_ksize=11)
参考文献
- https://numpy.org/doc/
- https://docs.opencv.org/master/
- https://opencv-python-tutroals.readthedocs.io/en/latest/py_tutorials/py_tutorials.html
- https://stackabuse.com/affine-image-transformations-in-python-with-numpy-pillow-and-opencv/
- https://cristianpb.github.io/blog/image-rotation-opencv
- 项目
标签: