Skip to content

Commit df09af0

Browse files
mingxingtancopybara-github
authored andcommitted
Add efficientdet-lite
PiperOrigin-RevId: 373276037
1 parent b2feae2 commit df09af0

File tree

1 file changed

+17
-0
lines changed
  • tensorflow_examples/lite/model_maker/third_party/efficientdet

1 file changed

+17
-0
lines changed

tensorflow_examples/lite/model_maker/third_party/efficientdet/README.md

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,16 @@
11
# EfficientDet
2+
[![Paper](http://img.shields.io/badge/Paper-arXiv.1911.09070-B3181B?logo=arXiv)](https://arxiv.org/abs/1911.09070)
3+
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.sandbox.google.com/github/google/automl/blob/master/efficientdet/tutorial.ipynb)
4+
[![TensorFlow Hub](https://img.shields.io/badge/TF%20Hub-Models-FF6F00?logo=tensorflow)](https://tfhub.dev/s?network-architecture=efficientdet)
5+
6+
27

38
[1] Mingxing Tan, Ruoming Pang, Quoc V. Le. EfficientDet: Scalable and Efficient Object Detection. CVPR 2020.
49
Arxiv link: https://arxiv.org/abs/1911.09070
510

611
Updates:
712

13+
- May10/2021: Added EfficientDet-lite checkpoints (by Yuqi and TFLite team)
814
- Mar25/2021: Added [Det-AdvProp](https://arxiv.org/abs/2103.13886) model checkpoints ([see this page](./Det-AdvProp.md)).
915
- Jul20/2020: Added keras/TF2 and new SOTA D7x: 55.1mAP with 153ms.
1016
- Apr22/2020: Sped up end-to-end latency: D0 has up to >200 FPS throughput on Tesla V100.
@@ -82,10 +88,21 @@ In addition, the following table includes a list of models trained with fixed 64
8288
| D5(640) [h5](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco640/efficientdet-d5-640.h5), [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco640/efficientdet-d5-640.tar.gz) | 46.6 | 26.6ms |
8389
| D6(640) [h5](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco640/efficientdet-d6-640.h5), [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco640/efficientdet-d6-640.tar.gz) | 47.9 | 33.8ms |
8490

91+
We have also provided a list of mobile-size lite models:
92+
93+
| Model | mAP (float) | Quantized mAP (int8) | Prameters | Mobile latency |
94+
| ------ | :------: | :------: | :------: | :------: |:------: |
95+
| lite0, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite0.tgz) | 26.41 | 26.10 | 4.3M | 36ms |
96+
| lite1, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite1.tgz) | 31.50 | 31.12 | 5.8M | 49ms |
97+
| lite2, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite2.tgz) | 35.06 | 34.69 | 7.2M | 69ms |
98+
| lite3, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite3.tgz) | 38.77 | 38.42 | 11M | 116ms |
99+
| lite3x, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite3x..gz) | 42.64 | 41.87 | 12M | 208ms |
100+
| lite4, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite4.tgz) | 43.18 | 42.83 | 20M | 260ms |
85101

86102

87103
## 3. Export SavedModel, frozen graph, tensort models, or tflite.
88104

105+
89106
Run the following command line to export models:
90107

91108
!rm -rf savedmodeldir

0 commit comments

Comments
 (0)