Adding MobileSSD float model to NN-API benchmark.
Approval for including models in AOSP given in b/115969705.
BUG: 113653409
Test: Covered by existing test: com.example.android.nn.benchmark.NNTest#testNNAPI10Seconds[ssd_mobilenet_v1_coco_float]
Change-Id: I1275bf268b4dc4a4d3ce2e116d96107f50542699
Merged-In: I1275bf268b4dc4a4d3ce2e116d96107f50542699
(cherry picked from commit 13b07434a84f1d1c0f876767a3afc2dea0c25ddf)
diff --git a/README.txt b/README.txt
index 7ec7681..b634f93 100644
--- a/README.txt
+++ b/README.txt
@@ -73,6 +73,19 @@
Apache License, Version 2.0
------------------------------------------------------------------
+- ssd_mobilenet_v1_coco_float.tflite
+Float version of MobileNet SSD tensorflow model based on:
+"Speed/accuracy trade-offs for modern convolutional object detectors."
+https://arxiv.org/abs/1611.10012
+Apache License, Version 2.0
+
+Generated from
+http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_2018_01_28.tar.gz
+on Sep 24 2018.
+See also: https://github.com/tensorflow/models/tree/master/research/object_detection
+Golden output generated with ToT tflite (Linux, x86_64 CPU).
+
+------------------------------------------------------------------
- ssd_mobilenet_v1_coco_quantized.tflite
8bit quantized MobileNet SSD tensorflow lite model based on:
"Speed/accuracy trade-offs for modern convolutional object detectors."
@@ -105,7 +118,7 @@
------------------------------------------------------------------
Input files:
------------------------------------------------------------------
-- ssd_mobilenet_v1_coco_quantized/tarmac.input
+- ssd_mobilenet_v1_coco_*/tarmac.input
Photo of airport tarmac by krtaylor@google.com, Apache License, Version 2.0
- cup_(128|160|192|224).input
Photo of cup by pszczepaniak@google.com, Apache License, Version 2.0
diff --git a/assets/models_list/test_models.json b/assets/models_list/test_models.json
index bf8d92d..418c1ca 100644
--- a/assets/models_list/test_models.json
+++ b/assets/models_list/test_models.json
@@ -180,6 +180,19 @@
]
},
{
+ "name": "ssd_mobilenet_v1_coco_float",
+ "testName": "SSD_MOBILENET_V1_COCO_FLOAT",
+ "baselineSec": 0.182,
+ "inputSize": [1, 300, 300, 3],
+ "dataSize": 4,
+ "inputOutputs": [
+ {
+ "input": "ssd_mobilenet_v1_coco_float/tarmac_float.input",
+ "output": "ssd_mobilenet_v1_coco_float/tarmac_float.output"
+ }
+ ]
+ },
+ {
"name": "ssd_mobilenet_v1_coco_quantized",
"testName": "SSD_MOBILENET_V1_COCO_QUANT8",
"baselineSec": 0.97,
diff --git a/assets/ssd_mobilenet_v1_coco_float.tflite b/assets/ssd_mobilenet_v1_coco_float.tflite
new file mode 100644
index 0000000..f57600b
--- /dev/null
+++ b/assets/ssd_mobilenet_v1_coco_float.tflite
Binary files differ
diff --git a/assets/ssd_mobilenet_v1_coco_float/tarmac_float.input b/assets/ssd_mobilenet_v1_coco_float/tarmac_float.input
new file mode 100644
index 0000000..7db39f8
--- /dev/null
+++ b/assets/ssd_mobilenet_v1_coco_float/tarmac_float.input
Binary files differ
diff --git a/assets/ssd_mobilenet_v1_coco_float/tarmac_float.output b/assets/ssd_mobilenet_v1_coco_float/tarmac_float.output
new file mode 100644
index 0000000..00719c5
--- /dev/null
+++ b/assets/ssd_mobilenet_v1_coco_float/tarmac_float.output
Binary files differ
