Skip to content

Commit d7c9f1b

Browse files
authored
Parameter count for EfficientDet lite models (#1138)
I think the `Prameters` count for `EfficientDet-lite` models are not correct: ```py # cd automl/efficientdet import hparams_config import inference from tf2 import efficientdet_keras config = hparams_config.get_efficientdet_config('efficientdet-lite0') model = efficientdet_keras.EfficientDetNet(config=config) # model = efficientdet_keras.EfficientDetModel(config=config) model.build((None, config.image_size, config.image_size, 3)) model.summary() ``` Summary results: | Model | Total params | Trainable params | | ------------------- | ------------ | ---------------- | | efficientdet-lite0 | 3,290,606 | 3,243,470 | | efficientdet-lite1 | 4,311,390 | 4,248,318 | | efficientdet-lite2 | 5,324,894 | 5,252,334 | | efficientdet-lite3 | 8,455,054 | 8,350,862 | | efficientdet-lite3x | 9,392,574 | 9,280,862 | | efficientdet-lite4 | 15,286,398 | 15,130,894 |
1 parent f53df57 commit d7c9f1b

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

efficientdet/README.md

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -113,12 +113,12 @@ We have also provided a list of mobile-size lite models:
113113

114114
| Model | mAP (float) | Quantized mAP (int8) | Prameters | Mobile latency |
115115
| ------ | :------: | :------: | :------: | :------: |
116-
| EfficientDet-lite0, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite0.tgz) | 26.41 | 26.10 | 4.3M | 36ms |
117-
| EfficientDet-lite1, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite1.tgz) | 31.50 | 31.12 | 5.8M | 49ms |
118-
| EfficientDet-lite2, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite2.tgz) | 35.06 | 34.69 | 7.2M | 69ms |
119-
| EfficientDet-lite3, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite3.tgz) | 38.77 | 38.42 | 11M | 116ms |
120-
| EfficientDet-lite3x, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite3x.tgz) | 42.64 | 41.87 | 12M | 208ms |
121-
| EfficientDet-lite4, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite4.tgz) | 43.18 | 42.83 | 20M | 260ms |
116+
| EfficientDet-lite0, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite0.tgz) | 26.41 | 26.10 | 3.2M | 36ms |
117+
| EfficientDet-lite1, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite1.tgz) | 31.50 | 31.12 | 4.2M | 49ms |
118+
| EfficientDet-lite2, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite2.tgz) | 35.06 | 34.69 | 5.3M | 69ms |
119+
| EfficientDet-lite3, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite3.tgz) | 38.77 | 38.42 | 8.4M | 116ms |
120+
| EfficientDet-lite3x, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite3x.tgz) | 42.64 | 41.87 | 9.3M | 208ms |
121+
| EfficientDet-lite4, [ckpt](https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-lite4.tgz) | 43.18 | 42.83 | 15.1M | 260ms |
122122

123123

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

0 commit comments

Comments
 (0)