【Detectron2】入门04-对测试数据做数据增强

在detectron2/data/transforms/transform.py文件中：

1、__all__=["想要添加的数据增加类名"]

2、定义数据增强类

class CorrupTransform(Transform):
    def __init__(self):
        super().__init__()

    def apply_image(self, img):
        
        '''
        要对图片做的操作
        '''
        return corrupted_img
    
    def apply_coords(self, coords)
        
        xxx

        return coords

    def inverse(self):
        return CorrupTransform()

3、在detectron2/modeling/test_time_augmentation/test_time_augmentation.py中，对DatasetMapperTTA类进行修改，添加新的数据增强方法

4、调用Trainer,test_with_TTA函数

在clean data上训练，corrpution data上测试，结果有显著下降。