Skip to content

运行AutoAugment类中example示例代码报错 #7248

Open
@vansin

Description

@vansin

背景:想通过代码链接链接文档中的example学习以下mmdet中的auto_augment

image

测试python程序如下

import mmdet.datasets.pipelines.auto_augment

from mmdet.datasets.pipelines import AutoAugment

import numpy as np

replace = (104, 116, 124)
policies = [
    [
        dict(type='Sharpness', prob=0.0, level=8),
        dict(
            type='Shear',
            prob=0.4,
            level=0,
            replace=replace,
            axis='x')
    ],
    [
        dict(
            type='Rotate',
            prob=0.6,
            level=10,
            replace=replace),
        dict(type='Color', prob=1.0, level=6)
    ]
]
augmentation = AutoAugment(policies)
img = np.ones(100, 100, 3)
gt_bboxes = np.ones(10, 4)
results = dict(img=img, gt_bboxes=gt_bboxes)
results = augmentation(results)

问题描述如下:

  1. Sharpness在mmdet中好像未实现(KeyError: 'Sharpness is not in the pipeline registry')
  2. 如下图片的replace=replace是什么意思呢,也报错了(TypeError: Shear: init() got an unexpected keyword argument 'replace')

image

环境信息

ubuntu 20.04
mmdet 2.21.0

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions