mmlab添加albumentations数据增强模块

系列文章目录

mmlab图像增强

---

---

前言

`提示：这里可以添加本文要记录的大概内容：

albumentation是深度学习经典的数据增强模块，本文介绍在mmlab采用albumentation自定义数据增强类

一、mmlab中自定义的数据变换类

该类型需要@TRANSFORMS.register_module()相应的进行修饰，并在跟数据变换的根目录中导入，才能在管道中使用。

二、使用步骤

1.参考代码

官方参考示例：

import random
import mmcv
from mmcv.transforms import BaseTransform, TRANSFORMS
#自定义类需要继承原有的BaseTransform类
@TRANSFORMS.register_module()
class MyFlip(BaseTransform):
    def __init__(self, direction: str):
        super().__init__()
        self.direction = direction
    def transform(self, results: dict) -> dict:
        img = results['img']
        results['img'] = mmcv