SeetaDet
WHAT's SeetaDet?
SeetaDet contains many useful object detectors, including R-CNN series, SSD, and the recent RetinaNet. We have achieved the same or higher performance than the baseline reported by the original paper.
This repository is based on our Dragon, while the style of codes is PyTorch. The torch-style codes help us to simplify the hierarchical pipeline of modern detection.
Installation
1. Install the required python packages
pip install cython pyyaml matplotlib
pip install opencv-python Pillow2. Compile the C Extensions
cd SeeTADet/compile
bash ./make.shResources
Pre-trained ImageNet models
| Model | Usage |
|---|---|
| VGG16.SSD | SSD |
| VGG16.RCNN | R-CNN |
| R-50.Affine | R-CNN, RetinaNet |
| R-101.Affine | R-CNN, RetinaNet |
| AirNet.SSD | SSD |
References
[1] Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks. Shaoqing Ren, Kaiming He, Ross Girshick, and Jian Sun. NIPS, 2015.
[2] Deep Residual Learning for Image Recognition. Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. CVPR, 2016.
[3] SSD: Single Shot MultiBox Detector. Wei Liu, Dragomir Anguelov, Dumitru Erhan, Christian Szegedy, Scott Reed, Cheng-Yang Fu, and Alexander C. Berg. ECCV, 2016.
[4] Feature Pyramid Networks for Object Detection. Tsung-Yi Lin, Piotr Dollár, Ross Girshick, Kaiming He, Bharath Hariharan, and Serge Belongie. CVPR, 2017.
[5] Focal Loss for Dense Object Detection. Tsung-Yi Lin, Priya Goyal, Ross Girshick, Kaiming He, and Piotr Dollár. ICCV, 2017.
[6] Mask R-CNN. Kaiming He, Georgia Gkioxari, Piotr Dollár and Ross Girshick. ICCV, 2017.
[7] Detectron. Ross Girshick, Ilija Radosavovic, Georgia Gkioxari, Piotr Dollar and Kaiming He. 2018.