aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorTreehugger Robot <treehugger-gerrit@google.com>2020-07-10 12:55:31 +0000
committerAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>2020-07-10 12:55:31 +0000
commit56702dc86cf64b07e305d7dd792b2ab50ca736df (patch)
tree047ed19c9866adfde3b9cdd4a3f6dc8753349000
parent5949fe0b88f46eb135b896bfc3b2cddf44e02775 (diff)
parentbefcd858a551acaed5bfe74bcfc1fa35de902ebf (diff)
downloadplatform_test_mlts_benchmark-56702dc86cf64b07e305d7dd792b2ab50ca736df.tar.gz
platform_test_mlts_benchmark-56702dc86cf64b07e305d7dd792b2ab50ca736df.tar.bz2
platform_test_mlts_benchmark-56702dc86cf64b07e305d7dd792b2ab50ca736df.zip
Merge "Add MobileNet v3 groups" am: 13fb073d2f am: befcd858a5
Original change: https://android-review.googlesource.com/c/platform/test/mlts/benchmark/+/1354604 Change-Id: Ica9bea5c6f6d2af14857352f5ab657bafc5e3dc2
-rwxr-xr-xresults/generate_result.py11
1 files changed, 9 insertions, 2 deletions
diff --git a/results/generate_result.py b/results/generate_result.py
index 1488c20..7358a51 100755
--- a/results/generate_result.py
+++ b/results/generate_result.py
@@ -28,6 +28,7 @@ import collections
import csv
import os
import re
+import math
class ScoreException(Exception):
@@ -54,6 +55,8 @@ KNOWN_GROUPS = [
(re.compile('mobilenet_v1.*'), 'MobileNet v1 Float'),
(re.compile('mobilenet_v2.*quant.*'), 'MobileNet v2 Quantized'),
(re.compile('mobilenet_v2.*'), 'MobileNet v2 Float'),
+ (re.compile('mobilenet_v3.*uint8.*'), 'MobileNet v3 Quantized'),
+ (re.compile('mobilenet_v3.*'), 'MobileNet v3 Float'),
(re.compile('tts.*'), 'LSTM Text-to-speech'),
(re.compile('asr.*'), 'LSTM Automatic Speech Recognition'),
]
@@ -146,9 +149,13 @@ def get_frequency_graph_min_max(results_with_bl):
def get_frequency_graph(time_freq_start_sec, time_freq_step_sec, time_freq_sec,
start_sec, end_sec):
"""Generate input x/y data for latency frequency graph."""
- left_to_pad = int((time_freq_start_sec - start_sec) / time_freq_step_sec)
+ left_to_pad = (int((time_freq_start_sec - start_sec) / time_freq_step_sec)
+ if time_freq_step_sec != 0
+ else math.inf)
end_time = time_freq_start_sec + len(time_freq_sec) * time_freq_step_sec
- right_to_pad = int((end_sec - end_time) / time_freq_step_sec)
+ right_to_pad = (int((end_sec - end_time) / time_freq_step_sec)
+ if time_freq_step_sec != 0
+ else math.inf)
# After pading more that 64 values, graphs start to look messy,
# bail out in that case.