开发面试题 Gitlab vim复制 控制跳转 CPU ios5 vue论坛 vue学习教程 linux内存管理 oracle删除表字段 webapp打包 java上传图片 python入门 python中open python如何调用函数 java环境配置 java初学 java文档 java读取文件内容 java中tostring方法 javac java声明变量 变量的类型 onenote2003 popen 东方头条邀请码 js轮播图代码 cmd代码 qq钱包实名认证 js正则匹配字符串 论文修改软件 羽毛球拍握法 流程图制作工具 ps平面广告设计教程 图片文字提取软件 sai怎么复制图层 伤害显示宏 ps镂空字体怎么做 音乐迷 pr如何抠图
当前位置: 首页 > 学习教程  > 编程语言

Albumentation使用指南

2020/8/31 14:03:39 文章标签:

Albumentation使用指南

import最好放在最前面否则可能会和其他模块冲突,import之后如果包OMP的错误就加两行

import os

os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'

 

资源

官方example

官方demo

 

使用

import albumentations as A
from PIL import Image
import cv2 as cv
import numpy as np
from data import get_dir, get_data, get_regr_tag

trans =  A.Compose(
    [
        A.HorizontalFlip(p = 0.5),
        # A.Transpose(),
        A.OneOf([
            A.IAAAdditiveGaussianNoise(),
            A.GaussNoise(),
        ], p=0.3),
        A.OneOf([
            A.MotionBlur(p=0.2),
            A.MedianBlur(blur_limit=3, p=0.1),
            A.Blur(blur_limit=3, p=0.1),
        ], p=0.5),
        A.ShiftScaleRotate(shift_limit=0.0625, scale_limit=0.2, rotate_limit=45, p=0.2),
        A.OneOf([
            A.OpticalDistortion(p=0.3),
            A.GridDistortion(p=0.1),
            A.IAAPiecewiseAffine(p=0.3),
        ], p=0.5),
        A.OneOf([
            A.CLAHE(clip_limit=2),
            A.IAASharpen(),
            A.IAAEmboss(),
            A.RandomBrightnessContrast(),
        ], p=0.5),
        A.HueSaturationValue(p=0.3),
    ])
src_data = get_data(f'D:/face_five', n_class = 5)
image = src_data[0][0]
img = trans(image = image)['image']

cv.imshow("da", img)
cv.waitKey(0)

A.OneOf是从列表里选一个相当于transforms里的randomCoice

A.Compose与transforms里的一样

 

API

随机调整亮度:

RandomBrightness(limit=2, p=0.5)

limit最大时现在的两倍,p是调整的概率

 

随机旋转:

Rotate(limit=89, p=0.5)

 

色度偏移:

RGBShift(r_shift_limit=133, g_shift_limit=146, b_shift_limit=26, p=0.5)

三原色在原本基础上增加或减小

 

伽马变换:

RandomGamma(gamma_limit=148, p=0.5)

色域的非线性变换

 

弹性变形:

ElasticTransform(alpha=155, sigma=210, alpha_affine=157, p=0.5)

 

光学变形:

OpticalDistortion(distort_limit=0.25, shift_limit=0.2, p=0.5)

distort_limit四周镜像填充宽度(图片大小不变)

shift_limit

 

垂直翻转:

VerticalFlip(p=0.5)

水平翻转:

HorizontalFlip(p=0.5)

 

随机90度旋转:

RandomRotate90(p = 0.5)

旋转平移缩放:

ShiftScaleRotate(shift_limit=0.8, scale_limit=1, rotate_limit=118, p=0.5)

rotate_limit:旋转角度限制

scale_limit:缩放尺寸限制,(0到2)

shift_limit:平移限制(0 到1)

周围空缺部分均镜面填补

 

中心裁剪:

CenterCrop(height=480, width=480, p=0.5)

 

图片压缩:

JpegCompression(quality_lower=80, quality_upper=100, p=0.5)

quality_lower&quality_upper设置压缩百分比的上下限

 

HSV偏移:

HueSaturationValue(hue_shift_limit=20, sat_shift_limit=20, val_shift_limit=27, p=0.5)

hue:色度

sat:饱和度

val:亮度

 

随机对比度:

RandomContrast(limit=2.3, p=0.5)

limit:0到3, 1为正常值

 

不定期更新。。


本文链接: http://www.dtmao.cc/news_show_150167.shtml

附件下载

相关教程

    暂无相关的数据...

共有条评论 网友评论

验证码: 看不清楚?