1. class 별로 map 결과 확인하고 싶을 때
config 파일에서 classwise=True 추가
test_evaluator = dict(
type='CocoMetric',
ann_file=data_root + 'annotations/instances_val2017.json',
metric='bbox',
format_only=False,
classwise=True, ## 이 부분 추가
backend_args=backend_args)
https://github.com/open-mmlab/mmdetection/issues/10813
2. tta 추가 하고 싶을 때
config 파일에 아래 코드 추가
'''
tta
'''
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'))]])
이후 test 수행할 때
python tools/test.py {config 경로} {checkpoint 경로} --tta
https://github.com/open-mmlab/mmdetection/pull/9452
'오답노트' 카테고리의 다른 글
[PIL] 이미지 불러올때 자동 회전 문제 발생 (0) | 2024.11.01 |
---|---|
[Colab] python 버전 확인 및 변경 (0) | 2024.10.30 |
[Error] bash: git: command not found (0) | 2024.10.28 |
vscode 브랜치 자동 넘어감 (0) | 2024.10.21 |