Skip to content

Commit

Permalink
Merge pull request ARM-software#4 from ARM-software/add_testing_data
Browse files Browse the repository at this point in the history
Added testing data for models, minor tweak to README
  • Loading branch information
tom-arm authored Dec 16, 2020
2 parents 18dcee2 + c8ece53 commit 2b941fe
Show file tree
Hide file tree
Showing 47 changed files with 144 additions and 2 deletions.
1 change: 1 addition & 0 deletions .gitattributes
Original file line number Diff line number Diff line change
Expand Up @@ -2,3 +2,4 @@
*.pb filter=lfs diff=lfs merge=lfs -text
*.data* filter=lfs diff=lfs merge=lfs -text
*.index filter=lfs diff=lfs merge=lfs -text
*.npy filter=lfs diff=lfs merge=lfs -text
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,8 @@ network:
network_parameters:
input_nodes:
- description: Single 224x224 RGB image with UINT8 values between 0 and 255
example_input:
path: models/image_classification/mobilenet_v2_1.0_224/tflite_uint8/testing_input/input
name: input
shape:
- 1
Expand All @@ -28,6 +30,7 @@ network_parameters:
shape:
- 1
- 1001
test_output_path: models/image_classification/mobilenet_v2_1.0_224/tflite_uint8/testing_output/output
operators:
TensorFlow Lite:
- ADD
Expand Down
Git LFS file not shown
Git LFS file not shown
3 changes: 3 additions & 0 deletions models/keyword_spotting/cnn_large/tflite_int8/definition.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@ network:
network_parameters:
input_nodes:
- description: The input is a processed MFCCs of shape (1, 490)
example_input:
path: models/keyword_spotting/cnn_large/tflite_int8/testing_input/input
name: input
shape:
- 1
Expand All @@ -27,6 +29,7 @@ network_parameters:
shape:
- 1
- 12
test_output_path: models/keyword_spotting/cnn_large/tflite_int8/testing_output/Identity
operators:
TensorFlow Lite:
- CONV_2D
Expand Down
Git LFS file not shown
Git LFS file not shown
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@ network:
network_parameters:
input_nodes:
- description: The input is a processed MFCCs of shape (1, 490)
example_input:
path: models/keyword_spotting/cnn_medium/tflite_int8/testing_input/input
name: input
shape:
- 1
Expand All @@ -27,6 +29,7 @@ network_parameters:
shape:
- 1
- 12
test_output_path: models/keyword_spotting/cnn_medium/tflite_int8/testing_output/Identity
operators:
TensorFlow Lite:
- CONV_2D
Expand Down
Git LFS file not shown
Git LFS file not shown
3 changes: 3 additions & 0 deletions models/keyword_spotting/cnn_small/tflite_int8/definition.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@ network:
network_parameters:
input_nodes:
- description: The input is a processed MFCCs of shape (1, 490)
example_input:
path: models/keyword_spotting/cnn_small/tflite_int8/testing_input/input
name: input
shape:
- 1
Expand All @@ -27,6 +29,7 @@ network_parameters:
shape:
- 1
- 12
test_output_path: models/keyword_spotting/cnn_small/tflite_int8/testing_output/Identity
operators:
TensorFlow Lite:
- CONV_2D
Expand Down
Git LFS file not shown
Git LFS file not shown
3 changes: 3 additions & 0 deletions models/keyword_spotting/dnn_large/tflite_int8/definition.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@ network:
network_parameters:
input_nodes:
- description: The input is a processed MFCCs of shape (1, 250)
example_input:
path: models/keyword_spotting/dnn_large/tflite_int8/testing_input/input
name: input
shape:
- 1
Expand All @@ -27,6 +29,7 @@ network_parameters:
shape:
- 1
- 12
test_output_path: models/keyword_spotting/dnn_large/tflite_int8/testing_output/Identity
operators:
TensorFlow Lite:
- DEQUANTIZE
Expand Down
Git LFS file not shown
Git LFS file not shown
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@ network:
network_parameters:
input_nodes:
- description: The input is a processed MFCCs of shape (1, 250)
example_input:
path: models/keyword_spotting/dnn_medium/tflite_int8/testing_input/input
name: input
shape:
- 1
Expand All @@ -27,6 +29,7 @@ network_parameters:
shape:
- 1
- 12
test_output_path: models/keyword_spotting/dnn_medium/tflite_int8/testing_output/Identity
operators:
TensorFlow Lite:
- DEQUANTIZE
Expand Down
Git LFS file not shown
Git LFS file not shown
3 changes: 3 additions & 0 deletions models/keyword_spotting/dnn_small/tflite_int8/definition.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@ network:
network_parameters:
input_nodes:
- description: The input is a processed MFCCs of shape (1, 250)
example_input:
path: models/keyword_spotting/dnn_small/tflite_int8/testing_input/input
name: input
shape:
- 1
Expand All @@ -27,6 +29,7 @@ network_parameters:
shape:
- 1
- 12
test_output_path: models/keyword_spotting/dnn_small/tflite_int8/testing_output/Identity
operators:
TensorFlow Lite:
- DEQUANTIZE
Expand Down
Git LFS file not shown
Git LFS file not shown
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@ network:
network_parameters:
input_nodes:
- description: The input is a processed MFCCs of shape (1, 490)
example_input:
path: models/keyword_spotting/ds_cnn_large/tflite_int8/testing_input/input
name: input
shape:
- 1
Expand All @@ -27,6 +29,7 @@ network_parameters:
shape:
- 1
- 12
test_output_path: models/keyword_spotting/ds_cnn_large/tflite_int8/testing_output/Identity
operators:
TensorFlow Lite:
- AVERAGE_POOL_2D
Expand Down
Git LFS file not shown
Git LFS file not shown
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@ network:
network_parameters:
input_nodes:
- description: The input is a processed MFCCs of shape (1, 490)
example_input:
path: models/keyword_spotting/ds_cnn_medium/tflite_int8/testing_input/input
name: input
shape:
- 1
Expand All @@ -27,6 +29,7 @@ network_parameters:
shape:
- 1
- 12
test_output_path: models/keyword_spotting/ds_cnn_medium/tflite_int8/testing_output/Identity
operators:
TensorFlow Lite:
- AVERAGE_POOL_2D
Expand Down
Git LFS file not shown
Git LFS file not shown
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@ network:
network_parameters:
input_nodes:
- description: The input is a processed MFCCs of shape (1, 490)
example_input:
path: models/keyword_spotting/ds_cnn_small/tflite_int8/testing_input/input
name: input
shape:
- 1
Expand All @@ -27,6 +29,7 @@ network_parameters:
shape:
- 1
- 12
test_output_path: models/keyword_spotting/ds_cnn_small/tflite_int8/testing_output/Identity
operators:
TensorFlow Lite:
- AVERAGE_POOL_2D
Expand Down
Git LFS file not shown
Git LFS file not shown
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,8 @@ network:
network_parameters:
input_nodes:
- description: A float input image.
example_input:
path: models/object_detection/ssd_mobilenet_v1/tflite_fp32/testing_input/normalized_input_image_tensor
name: normalized_input_image_tensor
shape:
- 1
Expand All @@ -30,15 +32,19 @@ network_parameters:
(y1, x1, y2, x2) scaled from 0 to 1.
name: TFLite_Detection_PostProcess
shape: []
test_output_path: models/object_detection/ssd_mobilenet_v1/tflite_fp32/testing_output/TFLite_Detection_PostProcess
- description: COCO detection classes for each object. 0=person, 10=fire hydrant.
name: TFLite_Detection_PostProcess:1
shape: []
test_output_path: models/object_detection/ssd_mobilenet_v1/tflite_fp32/testing_output/TFLite_Detection_PostProcess:1
- description: Detection scores for each object.
name: TFLite_Detection_PostProcess:2
shape: []
test_output_path: models/object_detection/ssd_mobilenet_v1/tflite_fp32/testing_output/TFLite_Detection_PostProcess:2
- description: The number of objects detected in each image.
name: TFLite_Detection_PostProcess:3
shape: []
test_output_path: models/object_detection/ssd_mobilenet_v1/tflite_fp32/testing_output/TFLite_Detection_PostProcess:3
operators:
TensorFlow Lite:
- CONCATENATION
Expand Down
Git LFS file not shown
Git LFS file not shown
Git LFS file not shown
Git LFS file not shown
Git LFS file not shown
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ Dataset: Coco Validation 2017
## Network Inputs
| Input Node Name | Shape | Description |
|-----------------|---------|-------------|
| image_tensor | (1, 300, 300, 3) | Input RGB images (a range of 0-255 per RGB channel). |
| normalized_input_image_tensor | (1, 300, 300, 3) | Input RGB images (a range of 0-255 per RGB channel). |

## Network Outputs
| Output Node Name | Shape | Description |
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,9 @@ network:
network_parameters:
input_nodes:
- description: Input RGB images (a range of 0-255 per RGB channel).
name: image_tensor
example_input:
path: models/object_detection/ssd_mobilenet_v1/tflite_uint8/testing_input/normalized_input_image_tensor
name: normalized_input_image_tensor
shape:
- 1
- 300
Expand All @@ -29,15 +31,19 @@ network_parameters:
- description: The y1, x1, y2, x2 coordinates of the bounding boxes for each detection
name: TFLite_Detection_PostProcess
shape: []
test_output_path: models/object_detection/ssd_mobilenet_v1/tflite_uint8/testing_output/TFLite_Detection_PostProcess
- description: The class of each detection
name: TFLite_Detection_PostProcess:1
shape: []
test_output_path: models/object_detection/ssd_mobilenet_v1/tflite_uint8/testing_output/TFLite_Detection_PostProcess:1
- description: The probability score for each classification
name: TFLite_Detection_PostProcess:2
shape: []
test_output_path: models/object_detection/ssd_mobilenet_v1/tflite_uint8/testing_output/TFLite_Detection_PostProcess:2
- description: A vector containing a number corresponding to the number of detections
name: TFLite_Detection_PostProcess:3
shape: []
test_output_path: models/object_detection/ssd_mobilenet_v1/tflite_uint8/testing_output/TFLite_Detection_PostProcess:3
operators:
TensorFlow Lite:
- CONCATENATION
Expand Down
Git LFS file not shown
Git LFS file not shown
Git LFS file not shown
Git LFS file not shown
Git LFS file not shown
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,8 @@ network:
network_parameters:
input_nodes:
- description: A 416x416 floating point input image.
example_input:
path: models/object_detection/yolo_v3_tiny/tflite_fp32/testing_input/inputs
name: inputs
shape:
- 1
Expand All @@ -33,6 +35,7 @@ network_parameters:
- 1
- 2535
- 85
test_output_path: models/object_detection/yolo_v3_tiny/tflite_fp32/testing_output/output_boxes
operators:
TensorFlow Lite:
- ADD
Expand Down
Git LFS file not shown
Git LFS file not shown

0 comments on commit 2b941fe

Please sign in to comment.