• [pytorch]数据集增强方法


    数据增强的库

    pytorch自带

    imgaug

    https://github.com/aleju/imgaug
    数据增强(旋转、翻转、加噪声、加对比度、亮度)
    (更多操作,也可以加天气效果,推荐)
    它可以实现的效果包括添加噪音、仿射变换、裁剪、翻转、旋转等,其效果图如下所示:

    Augmentor

    简单操作

    Albumentations

    是 2018年的一篇论文《Albumentations: fast and flexible image augmentations》的实现代码
    数据增强(添加下雨 下雪 雾效果)

    速度比大部分的库都要快;基于 numpy 和 OpenCV 两个库,并选择最合适的一个接口简单,灵活大量的多种的变换方法实现易于拓展应用到其他的任务或者库支持的变换操作对象有图像、masks、关键点、边界框支持 python 3.5-3.7 的版本可以和 PyTorch 结合使用已经被应用到很多深度学习的竞赛中,包括 Kaggle、topcoder,CVPR,MICCAI作者是 Kaggle Masters

    ImageEnhance

    适用于opencv-python、PIL
    同样来自一篇论文《Biomedical image augmentation using Augmentor》

    区分:

    imgaug:操作更多(但相对opencv-python,代码简洁),支持keypoint, bounding box同步处理 ******

    Albumentations: 据说可以进行 加雨雪雾的效果

    Augmentor:操作较少(但相对opencv-python,代码简介)

    ImageEnhance: 4种操作颜色增强

    1、对比度:白色画面(最亮时)下的亮度除以黑色画面(最暗时)下的亮度;
    2、色彩饱和度::彩度除以明度,指色彩的鲜艳程度,也称色彩的纯度;
    3、色调:向负方向调节会显现红色,正方向调节则增加黄色。适合对肤色对象进行微调;
    4、锐度:是反映图像平面清晰度和图像边缘锐利程度的一个指标。

    opencv-python:

    1、随机裁剪、旋转、翻转;
    2、随机颜色、明暗;
    3、仿射变换;

    数据增强的例子

    数据增强的论文

    Mixup
    来自 ICLR 2018 的论文:《Mixup: BEYOND EMPIRICAL RISK MINIMIZATION》

    Cutout
    2017年的论文:《Improved Regularization of Convolutional Neural Networks with Cutout》

    Cutmix
    ICCV 2019 的论文:《CutMix:Regularization Strategy to Train Strong Classifiers with Localizable Features》

    Augmix
    ICLR 2020 的论文:《AUGMIX: A SIMPLE DATA PROCESSING METHOD TO IMPROVE ROBUSTNESS AND UNCERTAINTY》

    fast-autoaugment
    NeurlIPS 2019 的论文《 Fast AutoAugment》

    AutoAugment
    CVPR 2019 的论文 《AutoAugment:Learning Augmentation Strategies from Data》

    RandAugment
    ICLR 2020 的论文《RandAugment: Practical automated data augmentation with a reduced search space》

    GridMask
    2020年的论文题目《GridMask Data Augmentation》

    imagecorruptions
    2019 年的论文《Benchmarking Robustness in Object Detection:Autonomous Driving when Winter is Coming》

    CycleGAN
    ICCV 2017 年的一篇论文《Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networkss》,也是 GAN 领域非常有名的一篇论文

    Small Object Augmentation
    2019 年的论文《Augmentation for small object detection》

  • 相关阅读:
    【FZYZOJ】细菌 题解(最短路)
    oracle 开发 第02章 查询
    oracle 开发 第01章 简介
    rhel配置163、epel、rpmforge的yum源
    linux 手动清除缓存
    查看linux内核版本和发行版本
    linux下安装rlwrap
    linux下安装vncserver
    Nagios 安装
    linux mutt发送报表
  • 原文地址:https://www.cnblogs.com/wildkid1024/p/16088909.html
Copyright © 2020-2023  润新知