[Feature] Support TTA

[zytx121]

open-mmlab/mmengine#611 is required.

Motivation

To support Test-Time-Aug (TTA).

Modification

• Add DetTTAModel inherited from the BaseTTAModel in MMEngine.
• Add --tta option in tools/test.py to enable TTA during the test.

Use cases

To use the default TTA (average the classification scores of images w/ and w/o flip)

 python tools/test.py configs/retinanet/retinanet_r50_fpn_1x_coco.py checkpoints/resnet50_coco.pth --tta

To use your own custom TTA method

1. Add tta_model and tta_pipeline in your config file:

tta_model = dict(
   type='DetTTAModel',
   tta_cfg=dict(nms=dict(type='nms', iou_threshold=0.5), max_per_img=100))
tta_pipeline = [
       dict(type='LoadImageFromFile', file_client_args=dict(backend='disk')),
       dict(type='TestTimeAug',
            transforms=[
                [dict(type='Resize', scale=(1333, 800), keep_ratio=True),
                 dict(type='Resize', scale=(1333, 600), keep_ratio=True)],
                [dict(type='RandomFlip', prob=1.),
                 dict(type='RandomFlip', prob=0.)],
                [dict(type='LoadAnnotations', with_bbox=True)],
                [dict(type='PackDetInputs',
                      meta_keys=('img_id', 'img_path', 'ori_shape',
                                 'img_shape', 'scale_factor', 'flip',
                                 'flip_direction'))]])
   ]

2. test with TTA

python tools/test.py configs/retinanet/retinanet_r50_fpn_1x_coco.py checkpoints.pth --tta

Checklist

1. Pre-commit or other linting tools are used to fix the potential lint issues.
2. The modification is covered by complete unit tests. If not, please add more unit test to ensure the correctness.
3. If the modification has potential influence on downstream projects, this PR should be tested with downstream projects, like MMDet or MMCls.
4. The documentation has been modified accordingly, like docstring or example tutorials.

[RangiLyu]

Remember to update the document of TTA after the usage is confirmed.

[hhaAndroid]

@zytx121 Please synchronize the latest code to ensure that ci can pass

[zytx121]

@zytx121 Please synchronize the latest code to ensure that ci can pass

Done.

[RangiLyu]

LGTM. It would be better if you can add a warning in test.py if there is no tta_cfg in the config.

[RangiLyu]

Add the use case to docs/en/user_guides/test.md

[zytx121]

Add the use case to docs/en/user_guides/test.md

LGTM. It would be better if you can add a warning in test.py if there is no tta_cfg in the config.

Done.