diff options
| author | Treehugger Robot <treehugger-gerrit@google.com> | 2020-07-10 12:42:51 +0000 |
|---|---|---|
| committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2020-07-10 12:42:51 +0000 |
| commit | befcd858a551acaed5bfe74bcfc1fa35de902ebf (patch) | |
| tree | e0a5a605ac53383a00a0946244fa7228c649698c | |
| parent | 97921ca5849354d8da3c79833887b8093c905bb4 (diff) | |
| parent | 13fb073d2f039221152e9257ef89d0d40242c3dc (diff) | |
| download | platform_test_mlts_benchmark-befcd858a551acaed5bfe74bcfc1fa35de902ebf.tar.gz platform_test_mlts_benchmark-befcd858a551acaed5bfe74bcfc1fa35de902ebf.tar.bz2 platform_test_mlts_benchmark-befcd858a551acaed5bfe74bcfc1fa35de902ebf.zip | |
Merge "Add MobileNet v3 groups" am: 13fb073d2f
Original change: https://android-review.googlesource.com/c/platform/test/mlts/benchmark/+/1354604
Change-Id: I3a4a7e04dd39c93245628f3658da1d65df3ba128
| -rwxr-xr-x | results/generate_result.py | 11 |
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. |