- FQDet: Fast-converging Query-based Detector by Cédric Picron, Punarjay Chakravarty and Tinne Tuytelaars.
01 Jun, 2023
: FQDet logs and checkpoints released.24 May, 2023
: FQDet code released (with configs and training scripts).23 Oct, 2022
: FQDet paper was accepted at NeurIPS 2022 (VTTA workshop).
Object detection results on the 2017 COCO validation set. The inference FLOPs and FPS are measured on the first 100 images of the 2017 COCO validation set using an NVIDIA GeForce RTX 3060 Ti GPU.
Backbone | Detector | Epochs | AP | Params | GFLOPs | FPS | Script | Log | Checkpoint |
---|---|---|---|---|---|---|---|---|---|
R50+FPN | FQDet | 12 | 43.3 | 33.9 M | 99.0 | 20.9 | script | log | checkpoint |
R50+TPN | FQDet | 12 | 45.5 | 42.2 M | 107.8 | 13.6 | script | log | checkpoint |
R50+DefEnc | FQDet | 12 | 47.2 | 44.1 M | 234.8 | 9.7 | script | log | checkpoint |
-
Environment:
- Install the
conda
package and environment management system if not already done. - Execute
source setup_env.sh
.
- Install the
-
Data preparation:
- Download the desired datasets.
- Modify the paths in
setup_data.sh
to point to your installation directories. - Execute
source setup_data.sh
.
-
Training: Execute
python main.py
with the desired command-line arguments. Some example training scripts, which were used to obtain the results from above, are found in thescripts
directory. -
Evalutation: Execute
python main.py --eval --eval_task $TASK
with the desired command-line arguments, with $TASK chosen from:- analysis: Analyze the computional cost of the given model.
- comparison: Compare the results from two different models.
- performance: Compute the model performance on the desired benchmark.
- profile: Profile the given model.
- visualize: Visualize the model predictions.
If you find FQDet useful in your research or wish to refer to the results published here, please use following BibTeX entry:
@article{picron2022fqdet,
title={FQDet: Fast-converging Query-based Detector},
author={Picron, C{\'e}dric and Chakravarty, Punarjay and Tuytelaars, Tinne},
journal={arXiv preprint arXiv:2210.02318},
year={2022}
}