signatrix / Efficientdet
Licence: mit
(Pretrained weights provided) EfficientDet: Scalable and Efficient Object Detection implementation by Signatrix GmbH
Stars: ✭ 565
Programming Languages
python
139335 projects - #7 most used programming language
Projects that are alternatives of or similar to Efficientdet
Tfjs Yolo Tiny
In-Browser Object Detection using Tiny YOLO on Tensorflow.js
Stars: ✭ 465 (-17.7%)
Mutual labels: object-detection
Yolo3 4 Py
A Python wrapper on Darknet. Compatible with YOLO V3.
Stars: ✭ 504 (-10.8%)
Mutual labels: object-detection
Yolov5 ncnn
🍅 Deploy NCNN on mobile phones. Support Android and iOS. 移动端NCNN部署,支持Android与iOS。
Stars: ✭ 535 (-5.31%)
Mutual labels: object-detection
Awesome Tiny Object Detection
🕶 A curated list of Tiny Object Detection papers and related resources.
Stars: ✭ 471 (-16.64%)
Mutual labels: object-detection
Yet Another Efficientdet Pytorch
The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
Stars: ✭ 4,945 (+775.22%)
Mutual labels: object-detection
Random Erasing
Random Erasing Data Augmentation. Experiments on CIFAR10, CIFAR100 and Fashion-MNIST
Stars: ✭ 511 (-9.56%)
Mutual labels: object-detection
Ssd tensorflow traffic sign detection
Implementation of Single Shot MultiBox Detector in TensorFlow, to detect and classify traffic signs
Stars: ✭ 459 (-18.76%)
Mutual labels: object-detection
Keras Yolov3 Mobilenet
I transfer the backend of yolov3 into Mobilenetv1,VGG16,ResNet101 and ResNeXt101
Stars: ✭ 552 (-2.3%)
Mutual labels: object-detection
Openvino Yolov3
YoloV3/tiny-YoloV3+RaspberryPi3/Ubuntu LaptopPC+NCS/NCS2+USB Camera+Python+OpenVINO
Stars: ✭ 500 (-11.5%)
Mutual labels: object-detection
Bmw Yolov4 Training Automation
This repository allows you to get started with training a state-of-the-art Deep Learning model with little to no configuration needed! You provide your labeled dataset or label your dataset using our BMW-LabelTool-Lite and you can start the training right away and monitor it in many different ways like TensorBoard or a custom REST API and GUI. NoCode training with YOLOv4 and YOLOV3 has never been so easy.
Stars: ✭ 533 (-5.66%)
Mutual labels: object-detection
Fpn
Feature Pyramid Networks for Object Detection
Stars: ✭ 485 (-14.16%)
Mutual labels: object-detection
Daily Paper Computer Vision
记录每天整理的计算机视觉/深度学习/机器学习相关方向的论文
Stars: ✭ 4,977 (+780.88%)
Mutual labels: object-detection
Tracking With Darkflow
Real-time people Multitracker using YOLO v2 and deep_sort with tensorflow
Stars: ✭ 515 (-8.85%)
Mutual labels: object-detection
Kmeans Anchor Boxes
k-means clustering with the Intersection over Union (IoU) metric as described in the YOLO9000 paper
Stars: ✭ 469 (-16.99%)
Mutual labels: object-detection
Curve Text Detector
This repository provides train&test code, dataset, det.&rec. annotation, evaluation script, annotation tool, and ranking.
Stars: ✭ 551 (-2.48%)
Mutual labels: object-detection
Keras Rcnn
Keras package for region-based convolutional neural networks (RCNNs)
Stars: ✭ 526 (-6.9%)
Mutual labels: object-detection
EfficientDet: Scalable and Efficient Object Detection
Introduction
Here is our pytorch implementation of the model described in the paper EfficientDet: Scalable and Efficient Object Detection paper (Note: We also provide pre-trained weights, which you could see at ./trained_models)
An example of our model's output.
Datasets
Dataset | Classes | #Train images | #Validation images |
---|---|---|---|
COCO2017 | 80 | 118k | 5k |
Create a data folder under the repository,
cd {repo_root}
mkdir data
-
COCO:
Download the coco images and annotations from coco website. Make sure to put the files as the following structure:
COCO ├── annotations │ ├── instances_train2017.json │ └── instances_val2017.json │── images ├── train2017 └── val2017
How to use our code
With our code, you can:
- Train your model by running python train.py
- Evaluate mAP for COCO dataset by running python mAP_evaluation.py
- Test your model for COCO dataset by running python test_dataset.py --pretrained_model path/to/trained_model
- Test your model for video by running python test_video.py --pretrained_model path/to/trained_model --input path/to/input/file --output path/to/output/file
Experiments
We trained our model by using 3 NVIDIA GTX 1080Ti. Below is mAP (mean average precision) for COCO val2017 dataset
Average Precision | IoU=0.50:0.95 | area= all | maxDets=100 | 0.314 |
---|---|---|---|---|
Average Precision | IoU=0.50 | area= all | maxDets=100 | 0.461 |
Average Precision | IoU=0.75 | area= all | maxDets=100 | 0.343 |
Average Precision | IoU=0.50:0.95 | area= small | maxDets=100 | 0.093 |
Average Precision | IoU=0.50:0.95 | area= medium | maxDets=100 | 0.358 |
Average Precision | IoU=0.50:0.95 | area= large | maxDets=100 | 0.517 |
Average Recall | IoU=0.50:0.95 | area= all | maxDets=1 | 0.268 |
Average Recall | IoU=0.50:0.95 | area= all | maxDets=10 | 0.382 |
Average Recall | IoU=0.50:0.95 | area= all | maxDets=100 | 0.403 |
Average Recall | IoU=0.50:0.95 | area= small | maxDets=100 | 0.117 |
Average Recall | IoU=0.50:0.95 | area= medium | maxDets=100 | 0.486 |
Average Recall | IoU=0.50:0.95 | area= large | maxDets=100 | 0.625 |
Results
Some predictions are shown below:
Requirements
- python 3.6
- pytorch 1.2
- opencv (cv2)
- tensorboard
- tensorboardX (This library could be skipped if you do not use SummaryWriter)
- pycocotools
- efficientnet_pytorch
References
- Mingxing Tan, Ruoming Pang, Quoc V. Le. "EfficientDet: Scalable and Efficient Object Detection." EfficientDet.
- Our implementation borrows some parts from RetinaNet.Pytorch
Citation
@article{EfficientDetSignatrix,
Author = {Signatrix GmbH},
Title = {A Pytorch Implementation of EfficientDet Object Detection},
Journal = {https://github.com/signatrix/efficientdet},
Year = {2020}
}
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].