TridentNet in Detectron2
Scale-Aware Trident Networks for Object Detection
Yanghao Li*, Yuntao Chen*, Naiyan Wang, Zhaoxiang Zhang
[TridentNet] [arXiv] [BibTeX]
In this repository, we implement TridentNet-Fast in Detectron2. Trident Network (TridentNet) aims to generate scale-specific feature maps with a uniform representational power. We construct a parallel multi-branch architecture in which each branch shares the same transformation parameters but with different receptive fields. TridentNet-Fast is a fast approximation version of TridentNet that could achieve significant improvements without any additional parameters and computational cost.
Training
To train a model, run
python /path/to/detectron2/projects/TridentNet/train_net.py --config-file <config.yaml>
For example, to launch end-to-end TridentNet training with ResNet-50 backbone on 8 GPUs, one should execute:
python /path/to/detectron2/projects/TridentNet/train_net.py --config-file configs/tridentnet_fast_R_50_C4_1x.yaml --num-gpus 8
Evaluation
Model evaluation can be done similarly:
python /path/to/detectron2/projects/TridentNet/train_net.py --config-file configs/tridentnet_fast_R_50_C4_1x.yaml --eval-only MODEL.WEIGHTS model.pth
Results on MS-COCO in Detectron2
| Model | Backbone | Head | lr sched | AP | AP50 | AP75 | APs | APm | APl | download |
|---|---|---|---|---|---|---|---|---|---|---|
| Faster | R50-C4 | C5-512ROI | 1X | 35.7 | 56.1 | 38.0 | 19.2 | 40.9 | 48.7 | model | metrics |
| TridentFast | R50-C4 | C5-128ROI | 1X | 38.0 | 58.1 | 40.8 | 19.5 | 42.2 | 54.6 | model | metrics |
| Faster | R50-C4 | C5-512ROI | 3X | 38.4 | 58.7 | 41.3 | 20.7 | 42.7 | 53.1 | model | metrics |
| TridentFast | R50-C4 | C5-128ROI | 3X | 40.6 | 60.8 | 43.6 | 23.4 | 44.7 | 57.1 | model | metrics |
| Faster | R101-C4 | C5-512ROI | 3X | 41.1 | 61.4 | 44.0 | 22.2 | 45.5 | 55.9 | model | metrics |
| TridentFast | R101-C4 | C5-128ROI | 3X | 43.6 | 63.4 | 47.0 | 24.3 | 47.8 | 60.0 | model | metrics |
Citing TridentNet
If you use TridentNet, please use the following BibTeX entry.
@InProceedings{li2019scale,
title={Scale-Aware Trident Networks for Object Detection},
author={Li, Yanghao and Chen, Yuntao and Wang, Naiyan and Zhang, Zhaoxiang},
journal={The International Conference on Computer Vision (ICCV)},
year={2019}
}