Skip to content
This repository was archived by the owner on Nov 25, 2023. It is now read-only.

Commit 597793a

Browse files
committed
2 parents eda3cf5 + 4e43a58 commit 597793a

File tree

2 files changed

+12
-6
lines changed

2 files changed

+12
-6
lines changed

models/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@ Original model stored for embedded-ai.bench.
1717
|squeezenetv1.1| fp32 |caffe| [prototxt](https://raw.githubusercontent.com/DeepScale/SqueezeNet/master/SqueezeNet_v1.1/deploy.prototxt), [caffemodel](https://github.com/DeepScale/SqueezeNet/raw/master/SqueezeNet_v1.1/squeezenet_v1.1.caffemodel) | [DeepScale/SqueezeNet](https://github.com/DeepScale/SqueezeNet/tree/master/SqueezeNet_v1.1) |
1818
|vgg16| fp32 |caffe| [prototxt](https://gist.githubusercontent.com/ksimonyan/211839e770f7b538e2d8/raw/0067c9b32f60362c74f4c445a080beed06b07eb3/VGG_ILSVRC_16_layers_deploy.prototxt), [caffemodel](http://www.robots.ox.ac.uk/~vgg/software/very_deep/caffe/VGG_ILSVRC_16_layers.caffemodel) | [ksimonyan](https://gist.github.com/ksimonyan/211839e770f7b538e2d8/) |
1919
|mobilenetv1| fp32 |tensorflow| [tflite&pb](https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_2018_02_22/mobilenet_v1_1.0_224.tgz) | [hosted_models](https://tensorflow.google.cn/lite/guide/hosted_models) |
20+
|mobilenetv1_quant| int8 |tensorflow| [tflite&pb](https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224_quant.tgz) | [hosted_models](https://tensorflow.google.cn/lite/guide/hosted_models) |
2021
|mobilenetv2| fp32 |tensorflow| [tflite&pb](https://storage.googleapis.com/download.tensorflow.org/models/tflite_11_05_08/mobilenet_v2_1.0_224.tgz) | [hosted_models](https://tensorflow.google.cn/lite/guide/hosted_models) |
22+
|mobilenetv2_quant| int8 |tensorflow| [tflite&pb](https://storage.googleapis.com/download.tensorflow.org/models/tflite_11_05_08/mobilenet_v2_1.0_224_quant.tgz) | [hosted_models](https://tensorflow.google.cn/lite/guide/hosted_models) |
2123
|squeezenetv1.1| fp32 |tensorflow| [tflite&pb](https://storage.googleapis.com/download.tensorflow.org/models/tflite/model_zoo/upload_20180427/squeezenet_2018_04_27.tgz) | [hosted_models](https://tensorflow.google.cn/lite/guide/hosted_models) |
2224
|vgg16| fp32 |tensorflow| [vgg_16_2016_08_28.tar.gz](http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz) | [tensorflow/models: Pre-trained Models](https://github.com/tensorflow/models/tree/master/research/slim#Pretrained) |

models/download.sh

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/usr/bin/env bash
2-
set -ex
2+
set -x
33

44
#alias wget_enhance='wget -c'
55
enable_quiet_wget=OFF
@@ -142,23 +142,27 @@ function rename_tensorflow_models() {
142142

143143
# squeezenet
144144
tar -zxvf squeezenet_2018_04_27.tgz
145-
mkdir tf_squeezenet_v1.1
146-
mv squeeze* ./tf_squeezenet_v1.1
145+
mkdir tf_squeezenetv1.1
146+
mv squeeze* ./tf_squeezenetv1.1
147+
cp tf_squeezenetv1.1/squeezenet.tflite tf_squeezenetv1.1/tf_squeezenetv1.1.tflite
147148

148149
# mobilenetv1 quant
149150
tar -zxvf mobilenet_v1_1.0_224_quant.tgz
150151
mkdir tf_mobilenetv1_quant
151152
mv mobilenet_v1_1.0_224_quant* ./tf_mobilenetv1_quant
153+
cp tf_mobilenetv1_quant/mobilenet_v1_1.0_224_quant.tflite tf_mobilenetv1_quant/tf_mobilenetv1_quant.tflite
152154

153155
# mobilenetv2 quant
154156
tar -zxvf mobilenet_v2_1.0_224_quant.tgz
155157
mkdir tf_mobilenetv2_quant
156158
mv mobilenet_v2_1.0_224_quant* ./tf_mobilenetv2_quant
159+
cp tf_mobilenetv2_quant/mobilenet_v2_1.0_224_quant.tflite tf_mobilenetv2_quant/tf_mobilenetv2_quant.tflite
157160

158161
# mnasnet 1.0 224
159-
tar -zxvf mnasnet_1.0_224_09_07_2018.tgz
160-
mkdir tf_mnasnet_1.0_224
161-
mv mnasnet* ./tf_mnasnet_1.0_224
162+
# TODO(ysh329): something wrong with not completed tar file
163+
#tar -zxvf mnasnet_1.0_224_09_07_2018.tgz
164+
#mkdir tf_mnasnet_1.0_224
165+
#mv mnasnet* ./tf_mnasnet_1.0_224
162166
}
163167

164168
function prepare_tensorflow_models() {

0 commit comments

Comments
 (0)