bit-bots / YOEO

YouOnlyEncodeOnce - A CNN for Embedded Object Detection and Semantic Segmentation
GNU General Public License v3.0
21 stars 4 forks source link

Class configuration file for defining which classes to group during nms #117

Closed phinik closed 7 months ago

phinik commented 8 months ago

Proposed changes

Added a class configuration file in which one can define which classes should be grouped during nms and what class name should be used instead when displaying the results. Grouped classes are further evaluated on a secondary metric.

Should replace the "multiple_robot_classes" argument by explicitly defining which classes are to be grouped.

Related issues

Necessary checks

Flova commented 7 months ago

Looks very good to me!