QueryDet-PyTorch
This repository is the official implementation of our paper: QueryDet: Cascaded Sparse Query for Accelerating High-Resolution Small Object Detection
Requirement
a. Install Pytorch 1.4 following here
b. Install APEX following here
c. Install our Pytorch based sparse convolution operation following here
d. Install the detectron2 toolkit following here, note that we build our approach based on version 0.2.1. Note you may follow the instructions to set COCO configs
d. Clone our repository and have fun with it!
Usage
1. Data preparation
a. To prepare MS-COCO, you may follow the instructions of Detectron2
b. We provide the data preprocessing code for VisDrone2018. You need to first download dataset from here
c. Check visdrone/data_prepare.py to process the dataset
2. Training
% train coco RetinaNet baseline
python train_coco.py --config-file models/retinanet/configs/coco/train.yaml --num-gpu 8 OUTPUT_DIR /path/to/workdir
% train coco QueryDet
python train_coco.py --config-file models/querydet/configs/coco/train.yaml --num-gpu 8 OUTPUT_DIR /path/to/workdir
% train VisDrone RetinaNet baseline
python train_visdrone.py --config-file models/retinanet/configs/visdrone/train.yaml --num-gpu 8 OUTPUT_DIR /path/to/workdir
% train VisDrone QueryDet
python train_visdrone.py --config-file models/querydet/configs/visdrone/train.yaml --num-gpu 8 OUTPUT_DIR /path/to/workdir
3. Test
% test coco RetinaNet baseline
python infer_coco.py --config-file models/retinanet/configs/coco/test.yaml --num-gpu 8 --eval-only MODEL.WEIGHTS /path/to/workdir/model_final.pth
% test coco QueryDet with Dense Inference
python infer_coco.py --config-file models/querydet/configs/coco/test.yaml --num-gpu 8 --eval-only MODEL.WEIGHTS /path/to/workdir/model_final.pth
% test coco QueryDet with CSQ
python infer_coco.py --config-file models/querydet/configs/coco/test.yaml --num-gpu 8 --eval-only MODEL.WEIGHTS /path/to/workdir/model_final.pth MODEL.QUERY.QUERY_INFER True
4 Apr 08, 2022
240 Dec 05, 2022
12 Dec 12, 2022
14 Sep 13, 2022
47 Nov 22, 2022
78 Jan 03, 2023
1.5k Jan 04, 2023
133 Jan 04, 2023
76 Dec 21, 2022
184 Dec 11, 2022
155 Dec 29, 2022
35 Dec 29, 2022
70 Dec 23, 2022
1.4k Dec 29, 2022
534 Dec 25, 2022
16 Oct 29, 2022
51 Jan 07, 2023
3 Oct 18, 2022
1.1k Dec 30, 2022
17 Dec 16, 2022