blob: e8fcfbd6ee0b441494a5d82cce86f5b7a58c16e3 [file] [log] [blame]
Przemyslaw Szczepaniak6a16f222018-05-18 13:40:18 +01001Copyright 2017 The Android Open Source Project
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is distributed on an "AS IS" BASIS,
11WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14------------------------------------------------------------------
15
16This directory contains models data for the Android Neural Networks API benchmarks.
17
18Included models:
19
20------------------------------------------------------------------
21- mobilenet_float.tflite
22MobileNet tensorflow lite model based on:
23"MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications"
24https://arxiv.org/abs/1704.04861
Przemyslaw Szczepaniak397b27f2018-06-04 16:00:25 +010025Apache License, Version 2.0
Przemyslaw Szczepaniak6a16f222018-05-18 13:40:18 +010026
Mika Raento38cb4be2018-08-01 13:26:44 +010027Downloaded from
28http://download.tensorflow.org/models/mobilenet_v1_2018_02_22/mobilenet_v1_1.0_224.tgz
29on Aug 1 2018 and converted using ToT toco.
30Golden output generated with ToT tensorflow (Linux, CPU).
31
Przemyslaw Szczepaniak6a16f222018-05-18 13:40:18 +010032------------------------------------------------------------------
33- mobilenet_quantized.tflite
348bit quantized MobileNet tensorflow lite model based on:
35"MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications"
36https://arxiv.org/abs/1704.04861
Przemyslaw Szczepaniak397b27f2018-06-04 16:00:25 +010037Apache License, Version 2.0
Przemyslaw Szczepaniak6a16f222018-05-18 13:40:18 +010038
Mika Raento38cb4be2018-08-01 13:26:44 +010039Downloaded from
40http://download.tensorflow.org/models/mobilenet_v1_2018_07_12/mobilenet_v1_1.0_224_quant.tgz
41on Aug 1 2018.
42Golden output generated with ToT tflite (Linux, CPU).
43
Przemyslaw Szczepaniak6a16f222018-05-18 13:40:18 +010044------------------------------------------------------------------
45- hdrnet_float.tflite
46Partial tensorflow lite model based on
47"Deep Bilateral Learningfor Real-Time Image Enhancement"
48https://groups.csail.mit.edu/graphics/hdrnet/
Przemyslaw Szczepaniak397b27f2018-06-04 16:00:25 +010049Apache License, Version 2.0
Przemyslaw Szczepaniak6a16f222018-05-18 13:40:18 +010050
51It's partial because Pack and Transpose operations were not supported at the time
52of model creation.
53
54------------------------------------------------------------------
55- hdrnet_quantized.tflite
568bit quantized partial tensorflow lite model based on
57"Deep Bilateral Learning for Real-Time Image Enhancement"
58https://groups.csail.mit.edu/graphics/hdrnet/
Przemyslaw Szczepaniak397b27f2018-06-04 16:00:25 +010059Apache License, Version 2.0
Przemyslaw Szczepaniak6a16f222018-05-18 13:40:18 +010060
Przemyslaw Szczepaniak918fba22018-05-25 15:15:48 +010061------------------------------------------------------------------
62- resnet_float.tflite
63ResNet tensorflow lite model based on:
64"Deep Residual Learning for Image Recognition"
65https://arxiv.org/abs/1512.03385
Przemyslaw Szczepaniak397b27f2018-06-04 16:00:25 +010066Apache License, Version 2.0
Przemyslaw Szczepaniak918fba22018-05-25 15:15:48 +010067
68------------------------------------------------------------------
69- resnet_quantized.tflite
708bit quantized ResNet tensorflow lite model based on:
71"Deep Residual Learning for Image Recognition"
72https://arxiv.org/abs/1512.03385
Przemyslaw Szczepaniak397b27f2018-06-04 16:00:25 +010073Apache License, Version 2.0
74
Viet Danga1d44e82018-07-04 13:35:37 +010075------------------------------------------------------------------
76- tts_float.tflite
77TTS tensorflow lite model based on:
78"Fast, Compact, and High Quality LSTM-RNN Based Statistical Parametric Speech Synthesizers for
79Mobile Devices"
80https://ai.google/research/pubs/pub45379
81Apache License, Version 2.0
82
83Note that the tensorflow lite model is the acoustic model in the paper. It is used because it is
84much heavier than the duration model.
Przemyslaw Szczepaniak397b27f2018-06-04 16:00:25 +010085
86------------------------------------------------------------------
87Input files:
88------------------------------------------------------------------
89- cup.input
90Photo of cup by pszczepaniak@google.com, Apache License, Version 2.0
Viet Dang70697032018-09-07 16:25:18 +010091- tts_float/arctic_*.input
92Linguistic features and durations generated from text sentences from the CMU Arctic set
93(http://www.festvox.org/cmu_arctic/cmuarctic.data), Apache License, Version 2.0
Przemyslaw Szczepaniak397b27f2018-06-04 16:00:25 +010094------------------------------------------------------------------
95
Przemyslaw Szczepaniak918fba22018-05-25 15:15:48 +010096
Przemyslaw Szczepaniak6a16f222018-05-18 13:40:18 +010097TODO(pszczepaniak): Update hdrnet to full model with pack and transpose
98TODO(pszczepaniak): Provide at least 5 inputs outputs for each model
99