[pytorch] 图像识别之augmix

MachineLP

发布于 2020-02-18 15:58:55

6.2K00

代码可运行

文章被收录于专栏：小鹏的专栏小鹏的专栏

运行总次数：0

代码可运行

augmix: https://github.com/google-research/augmix

论文：https://arxiv.org/abs/1912.02781

效果展示：

使用方法：

import augmentations
import numpy as np
from PIL import Image

def apply_op(image, op, severity):
    image = np.clip(image * 255., 0, 255).astype(np.uint8)
    pil_img = Image.fromarray(image)  # Convert to PIL.Image
    pil_img = op(pil_img, severity)
    return np.asarray(pil_img) / 255.

def augment_and_mix(image, severity=3, width=3, depth=-1, alpha=1.):
    """Perform AugMix augmentations and compute mixture.
    Args:
      image: Raw i

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

https