码迷,mamicode.com
首页 > 其他好文 > 详细

数据扩增(data augmentation)

时间:2020-02-20 13:13:02      阅读:298      评论:0      收藏:0      [点我收藏+]

标签:cal   ice   imwrite   ota   jpeg   github   str   nta   and   

能够查阅的网址:

https://github.com/CrazyVertigo/awesome-data-augmentation

这个网址包含了很多主流的数据扩增方法。涉及数据扩增,建议查阅这个网址,自己去找想用方法。非常nice的是,怎么安装、调用,都有非常详细的说明。

本人采用过的数据扩增方法1:

    参考的网址:https://github.com/xtj49/imgaug

安装:

     pip install albumentations

示例调用代码:

import albumentations as albu

def train_transform(image, p=1):
   aug = albu.Compose([albu.RandomRotate90(p=0.5),
                       albu.HorizontalFlip(p=0.5),
                       albu.RandomGamma(gamma_limit=(80, 120), p=0.5),
                       albu.JpegCompression(quality_lower=70, quality_upper=90, p=0.5),
                       albu.RandomScale(scale_limit=(0.5, 2), interpolation=cv2.INTER_CUBIC, p=1)
                       ], p=p)
   return aug(image=image)[image]

import cv2
import pdb;pdb.set_trace()

img=cv2.imread(0.jpg)
I=train_transform(img, p=1)
cv2.imwrite(1.jpg,I)

 

 

   

 

数据扩增(data augmentation)

标签:cal   ice   imwrite   ota   jpeg   github   str   nta   and   

原文地址:https://www.cnblogs.com/wjjcjj/p/12335040.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!