Awesome
Box-DETR: Understanding and Boxing Conditional Spatial Queries
The code is comming soon.
Code for the paper Box-DETR: Understanding and Boxing Conditional Spatial Queries.
Modified from DAB-DETR (https://github.com/IDEA-Research/DAB-DETR)
Installation
This code is tested on Python 3.8 and Pytorch 1.9.0
1.Install Pytorch and torchvision
Follow the instruction on https://pytorch.org/get-started/locally/.
2.Install other needed packages
pip install -r requirements.txt
Train the model
Run (8 GPUs) Box-DETR-R50 by
cd Box-DETR
sh train.sh
For running with 4 GPUs, please modify --nproc_per_node' to 4 and
--batch_size' to 4.
For other settings, please add `--backbone' or '--num_patterns'.