inference
GitHub
官網(wǎng)文檔
csdn blog 各功能中文介紹
安裝
pip install imgaug
使用示例
imgaug的基本使用流程：
定義變換序列（Sequential）→ 讀入圖片（imread）→ 執(zhí)行變換（augment_images）→ 保存圖片（imwrite）
# coding:utf-8
# 示例代碼
import cv2
from imgaug import augmenters as iaa
seq = iaa.Sequential([
iaa.Sometimes(0.5, iaa.Crop(px=(0,5))), # 以0.5的概率侮措，從每側(cè)裁剪圖像0到5px（隨機(jī)選擇）
iaa.Fliplr(0.5), # 0.5概率水平翻轉(zhuǎn)圖像
iaa.GaussianBlur(sigma=(0, 3.0)) # 使用0到3.0的sigma模糊圖像
iaa.Resize((0.5, 1.5)), # 將w和h在0.5-1.5倍范圍內(nèi)resize
])
# 單張圖像增強(qiáng)10次
img = cv2.imread('27.bmp')
for i in range(10):
img_aug = seq.augment_image(img)
cv2.imwrite('1.bmp', img_aug)
# 批量圖像增強(qiáng)泣洞，每張圖像增強(qiáng)5次
img_list = [cv2.imread('1.bmp'), cv2.imread('2.bmp')]
for i in range(5):
imgs_aug = seq.augment_images(img_list)
for j, img in enumerate(imgs_aug):
cv2.imwrite('str(i)+'_'+str(j)+'.bmp, img)
