Cascaded Sparse Query for Accelerating High-Resolution Small Object Detection
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
b. Install APEX for mixed precision training
c. Install our Pytorch based sparse convolution toolkit
d. Install the detectron2 toolkit. Note we build our approach based on version 0.2.1, you may follow the instructions to set environment configs
e. Install the Detectron2_Backbone for usage of MobileNet and ShuffleNet
f. 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