[mmdetection] 탐지 개체수 설정방법

mmdetection에는 이미지 하나당 감지될 수 있는 개체수가 정해져있다. (default 100)

내가 다루는 데이터의 경우 하나의 이미지에 보통 2~300개의 instance가 존재하므로 위 설정을 바꿀 필요가 있었다. 

 

변경방법은 다음과 같다. 

 

mmdetection/configs/_base_/models/mask_rcnn_r50_fpn.py

 

위 파일에서 맨 아래부분의 test_cfg 부분을 살펴보면

 rcnn 내부에 max_per_img 이 변수가 하나의 이미지에서 탐지할 수 있는 최대 개체수 파라미터다. 

해당 파라미터를 1000으로 변경하면 최대 1000개까지 instance를 탐지할 수 있다.

test_cfg=dict(
    rpn=dict(
        nms_pre=1000,
        max_per_img=1000,
        nms=dict(type='nms', iou_threshold=0.7),
        min_bbox_size=0),
    rcnn=dict(
        score_thr=0.05,
        nms=dict(type='nms', iou_threshold=0.5),
        max_per_img=100,  # the maximum number of detectable objects
        mask_thr_binary=0.5)))

 

이상!