Skip to content

Latest commit

 

History

History
74 lines (51 loc) · 2.68 KB

ssdlite_mobilenet_v2.md

File metadata and controls

74 lines (51 loc) · 2.68 KB
Raw

ssdlite_mobilenet_v2

Use Case and High-Level Description

The ssdlite_mobilenet_v2 model is used for object detection. For details, see the paper, MobileNetV2: Inverted Residuals and Linear Bottlenecks.

Specification

Metric Value
Type Detection
GFLOPs 1.525
MParams 4.475
Source framework TensorFlow*

Accuracy

Metric Value
coco_precision 24.2946%

Input

Original Model

Image, name: image_tensor, shape: [1x300x300x3], format: [BxHxWxC], where:

- B - batch size
- H - image height
- W - image width
- C - number of channels

Expected color order: RGB.

Converted Model

Image, name: image_tensor, shape: [1x3x300x300], format [BxCxHxW], where:

  • B - batch size
  • C - number of channels
  • H - image height
  • W - image width

Expected color order: BGR.

Output

Original Model

  1. Classifier, name: detection_classes. Contains predicted bounding-boxes classes in a range [1, 91]. The model was trained on Microsoft* COCO dataset version with 90 categories of object, 0 class is for background.
  2. Probability, name: detection_scores. Contains probability of detected bounding boxes.
  3. Detection box, name: detection_boxes. Contains detection boxes coordinates in format [y_min, x_min, y_max, x_max], where (x_min, y_min) are coordinates of the top left corner, (x_max, y_max) are coordinates of the right bottom corner. Coordinates are rescaled to input image size.
  4. Detections number, name: num_detections. Contains the number of predicted detection boxes.

Converted Model

The array of summary detection information, name: DetectionOutput, shape: [1, 1, N, 7], where N is the number of detected bounding boxes. For each detection, the description has the format: [image_id, label, conf, x_min, y_min, x_max, y_max], where:

  • image_id - ID of the image in the batch
  • label - predicted class ID
  • conf - confidence for the predicted class
  • (x_min, y_min) - coordinates of the top left bounding box corner (coordinates are stored in a normalized format, in a range [0, 1])
  • (x_max, y_max) - coordinates of the bottom right bounding box corner (coordinates are stored in a normalized format, in a range [0, 1])

Legal Information

The original model is distributed under the Apache License, Version 2.0. A copy of the license is provided in APACHE-2.0-TensorFlow.txt.