OW-DETR: Open-world Detection Transformer (CVPR 2022)
[Paper]
Akshita Gupta*, Sanath Narayan*, K J Joseph, Salman Khan, Fahad Shahbaz Khan, Mubarak Shah
(
Introduction
Open-world object detection (OWOD) is a challenging computer vision problem, where the task is to detect a known set of object categories while simultaneously identifying unknown objects. Additionally, the model must incrementally learn new classes that become known in the next training episodes. Distinct from standard object detection, the OWOD setting poses significant challenges for generating quality candidate proposals on potentially unknown objects, separating the unknown objects from the background and detecting diverse unknown objects. Here, we introduce a novel end-to-end transformer-based framework, OW-DETR, for open-world object detection. The proposed OW-DETR comprises three dedicated components namely, attention-driven pseudo-labeling, novelty classification and objectness scoring to explicitly address the aforementioned OWOD challenges. Our OW-DETR explicitly encodes multi-scale contextual information, possesses less inductive bias, enables knowledge transfer from known classes to the unknown class and can better discriminate between unknown objects and background. Comprehensive experiments are performed on two benchmarks: MS-COCO and PASCAL VOC. The extensive ablations reveal the merits of our proposed contributions. Further, our model outperforms the recently introduced OWOD approach, ORE, with absolute gains ranging from $1.8%$ to $3.3%$ in terms of unknown recall on MS-COCO. In the case of incremental object detection, OW-DETR outperforms the state-of-the-art for all settings on PASCAL VOC.
Installation
Requirements
We have trained and tested our models on Ubuntu 16.0, CUDA 10.2, GCC 5.4, Python 3.7
conda create -n owdetr python=3.7 pip
conda activate owdetr
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
pip install -r requirements.txt
Compiling CUDA operators
cd ./models/ops
sh ./make.sh
# unit test (should see all checking is True)
python test.py
Dataset & Results
OWOD proposed splits
The splits are present inside data/VOC2007/OWOD/ImageSets/ folder. The remaining dataset can be downloaded using this link
The files should be organized in the following structure:
OW-DETR/
βββ data/
βββ VOC2007/
βββ OWOD/
βββ JPEGImages
βββ ImageSets
βββ Annotations
Results
| Task1 | Task2 | Task3 | Task4 | ||||
|---|---|---|---|---|---|---|---|
| Method | U-Recall | mAP | U-Recall | mAP | U-Recall | mAP | mAP |
| ORE-EBUI | 4.9 | 56.0 | 2.9 | 39.4 | 3.9 | 29.7 | 25.3 |
| OW-DETR | 7.5 | 59.2 | 6.2 | 42.9 | 5.7 | 30.8 | 27.8 |
Our proposed splits
The splits are present inside data/VOC2007/OWDETR/ImageSets/ folder. The remaining dataset can be downloaded using this link
The files should be organized in the following structure:
OW-DETR/
βββ data/
βββ VOC2007/
βββ OWDETR/
βββ JPEGImages
βββ ImageSets
βββ Annotations
Currently, Dataloader and Evaluator followed for OW-DETR is in VOC format.
Results
| Task1 | Task2 | Task3 | Task4 | ||||
|---|---|---|---|---|---|---|---|
| Method | U-Recall | mAP | U-Recall | mAP | U-Recall | mAP | mAP |
| ORE-EBUI | 1.5 | 61.4 | 3.9 | 40.6 | 3.6 | 33.7 | 31.8 |
| OW-DETR | 5.7 | 71.5 | 6.2 | 43.8 | 6.9 | 38.5 | 33.1 |
Training
Training on single node
To train OW-DETR on a single node with 8 GPUS, run
./run.sh
Training on slurm cluster
To train OW-DETR on a slurm cluster having 2 nodes with 8 GPUS each, run
sbatch run_slurm.sh
Evaluation
For reproducing any of the above mentioned results please run the run_eval.sh file and add pretrained weights accordingly.
Note: For more training and evaluation details please check the Deformable DETR reposistory.
License
This repository is released under the Apache 2.0 license as found in the LICENSE file.
Citation
If you use OW-DETR, please consider citing:
@inproceedings{gupta2021ow,
title={OW-DETR: Open-world Detection Transformer},
author={Gupta, Akshita and Narayan, Sanath and Joseph, KJ and
Khan, Salman and Khan, Fahad Shahbaz and Shah, Mubarak},
booktitle={CVPR},
year={2022}
}
Contact
Should you have any question, please contact
Acknowledgments:
OW-DETR builds on previous works code base such as Deformable DETR, Detreg, and OWOD. If you found OW-DETR useful please consider citing these works as well.
336 Dec 25, 2022
489 Jan 7, 2023
87 Jan 8, 2023
64 Jan 7, 2023
365 Dec 30, 2022
182 Dec 23, 2022
430 Dec 23, 2022
36 Sep 13, 2022
281 Dec 30, 2022
141 Dec 30, 2022
761 Dec 26, 2022
153 Dec 2, 2022
133 Dec 22, 2022
58 Dec 28, 2022
13 Nov 29, 2022
3 Aug 20, 2022
37 Nov 27, 2022
2k Jan 5, 2023
99 Dec 6, 2022