SSD MobileNet v1 is a object detection network, that localizes and identifies objects in an input image. This is a TF Lite quantized version that takes a 300x300 input image and outputs detections for this image. This model is trained and quantized by Google.
Apache-2.0
The class labels associated with this model can be downloaded by running the script get_class_labels.sh.
Code to recreate this model can be found here.
Dataset: Coco Validation 2017
| Platform |
Optimized |
| Cortex-A |
✖️ |
| Cortex-M |
✖️ |
| Mali GPU |
✔️ |
| Ethos U |
✖️ |
- ✔️ - Will run on this platform.
- ✖️ - Will not run on this platform.
| Optimization |
Value |
| Quantization |
UINT8 |
| Input Node Name |
Shape |
Description |
| normalized_input_image_tensor |
(1, 300, 300, 3) |
Input RGB images (a range of 0-255 per RGB channel). |
| Output Node Name |
Shape |
Description |
| TFLite_Detection_PostProcess |
() |
The y1, x1, y2, x2 coordinates of the bounding boxes for each detection |
| TFLite_Detection_PostProcess:1 |
() |
The class of each detection |
| TFLite_Detection_PostProcess:2 |
() |
The probability score for each classification |
| TFLite_Detection_PostProcess:3 |
() |
A vector containing a number corresponding to the number of detections |