QueryDet: Cascaded Sparse Query for Accelerating High-Resolution SmallObject Detection

Last update: Dec 31, 2022

Related tags

Overview

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

QueryDet: Cascaded Sparse Query for Accelerating High-Resolution SmallObject Detection

Related tags

Overview

QueryDet-PyTorch

Requirement

Usage

1. Data preparation

2. Training

3. Test

Owner

Chenhongyi Yang

This repository contains a toolkit for collecting, labeling and tracking object keypoints

3rd Place Solution for ICCV 2021 Workshop SSLAD Track 3A - Continual Learning Classification Challenge

Re-implememtation of MAE (Masked Autoencoders Are Scalable Vision Learners) using PyTorch.

Rotation Robust Descriptors

3ds-Ghidra-Scripts - Ghidra scripts to help with 3ds reverse engineering

a short visualisation script for pyvideo data

GrabGpu_py: a scripts for grab gpu when gpu is free

Lightweight tool to perform MITM attack on local network

A human-readable PyTorch implementation of "Self-attention Does Not Need O(n^2) Memory"

The official repo for OC-SORT: Observation-Centric SORT on video Multi-Object Tracking. OC-SORT is simple, online and robust to occlusion/non-linear motion.

Learning Versatile Neural Architectures by Propagating Network Codes

source code the paper Fast and Robust Iterative Closet Point.

Rainbow: Combining Improvements in Deep Reinforcement Learning

MWPToolkit is a PyTorch-based toolkit for Math Word Problem (MWP) solving.

Neural network-based build time estimation for additive manufacturing

Efficient Conformer: Progressive Downsampling and Grouped Attention for Automatic Speech Recognition

This is a work in progress reimplementation of Instant Neural Graphics Primitives

The official repo of the CVPR2021 oral paper: Representative Batch Normalization with Feature Calibration

Python lib to talk to pylontech lithium batteries (US2000, US3000, ...) using RS485

Generic U-Net Tensorflow implementation for image segmentation