Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Simple multi-dataset detection

About

How do we build a general and broad object detection system? We use all labels of all concepts ever annotated. These labels span diverse datasets with potentially inconsistent taxonomies. In this paper, we present a simple method for training a unified detector on multiple large-scale datasets. We use dataset-specific training protocols and losses, but share a common detection architecture with dataset-specific outputs. We show how to automatically integrate these dataset-specific outputs into a common semantic taxonomy. In contrast to prior work, our approach does not require manual taxonomy reconciliation. Experiments show our learned taxonomy outperforms a expert-designed taxonomy in all datasets. Our multi-dataset detector performs as well as dataset-specific models on each training domain, and can generalize to new unseen dataset without fine-tuning on them. Code is available at https://github.com/xingyizhou/UniDet.

Xingyi Zhou, Vladlen Koltun, Philipp Kr\"ahenb\"uhl• 2021

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)
AP45.5
2454
Instance SegmentationCOCO 2017 (val)--
1144
Instance SegmentationCityscapes (val)--
239
Object DetectionCrowdHuman (val)--
52
Object DetectionObjects365 (val)
mAP33.7
48
Instance SegmentationScanNet (val)--
39
Object DetectionCityscapes (val)
mAP500.526
31
Object DetectionObject365
AP33.7
17
Object DetectionScanNet (val)
mAP@0.532.2
15
Object DetectionPASCAL VOC (val)
mAP5083.1
11
Showing 10 of 27 rows

Other info

Code

Follow for update