facebookresearch / Detectron

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.
Apache License 2.0
26.21k stars 5.45k forks source link

How to limit detections to one object per image? #951

Open manyaafonso opened 4 years ago

manyaafonso commented 4 years ago

Expected results

It is assumed that for my data (single class), at the most one object can and should be detected per image.

Actual results

Multiple objects of the same class are detected in some images.

Detailed steps to reproduce

infer_simple.py

System information

Is there any setting or parameters to force MaskRCNN to learn that there can be at most one object of the single class of interest per image?