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

MoveNetを用いたPythonでの姿勢推定のデモ

Official implementation of the paper "Lightweight Deep CNN for Natural Image Matting via Similarity Preserving Knowledge Distillation"

Aspect-Sentiment-Multiple-Opinion Triplet Extraction (NLPCC 2021)

Multi-Scale Geometric Consistency Guided Multi-View Stereo

Myia prototyping

A Genetic Programming platform for Python with TensorFlow for wicked-fast CPU and GPU support.

Non-Vacuous Generalisation Bounds for Shallow Neural Networks

Punctuation Restoration using Transformer Models for High-and Low-Resource Languages

FocusFace: Multi-task Contrastive Learning for Masked Face Recognition

Few-NERD: Not Only a Few-shot NER Dataset

Safe Policy Optimization with Local Features

MonoScene: Monocular 3D Semantic Scene Completion

CLIPImageClassifier wraps clip image model from transformers

Open-source code for Generic Grouping Network (GGN, CVPR 2022)

Implementation of Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learning

Hashformers is a framework for hashtag segmentation with transformers.

ivadomed is an integrated framework for medical image analysis with deep learning.

[ICCV'21] UNISURF: Unifying Neural Implicit Surfaces and Radiance Fields for Multi-View Reconstruction

This is implementation of AlexNet(2012) with 3D Convolution on TensorFlow (AlexNet 3D).

CS506-Spring2022 - Code and Slides for Boston University CS 506