Skip to content

Commit bbd1c07

Browse files
authored
Merge pull request #141 from slahiruk/Efficientnet-lite
EfficientNet lite dependency. Thanks to @slahiruk for the contribution!
2 parents 49b2175 + 304c7bc commit bbd1c07

File tree

6 files changed

+62
-0
lines changed

6 files changed

+62
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
4f99bd653a4cf94b

script/.cm/alias-u-4f99bd653a4cf94b

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
install-package-tensorflowmodel-efficientnet-lite
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
{}
Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
{
2+
"backup_data_uid": "4f99bd653a4cf94b",
3+
"backup_module_uid": "84e27ad9dd12e734",
4+
"backup_module_uoa": "script",
5+
"control": {
6+
"engine": "CK",
7+
"iso_datetime": "2020-07-27T17:07:19.766100",
8+
"version": [
9+
"1",
10+
"11",
11+
"4",
12+
"1"
13+
]
14+
},
15+
"data_name": "install-package-tensorflowmodel-efficientnet-lite"
16+
}
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
{}
Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
#! /bin/bash
2+
3+
#
4+
# Copyright (c) 2018-2020 cTuning foundation.
5+
# See CK COPYRIGHT.txt for copyright details.
6+
#
7+
# SPDX-License-Identifier: BSD-3-Clause.
8+
# See CK LICENSE.txt for licensing details.
9+
#
10+
# EfficientNet-Lite for TensorFlow
11+
# Python model and weights install script
12+
13+
########################################################################
14+
echo
15+
echo "Download weights from ${PACKAGE_URL} ..."
16+
wget ${PACKAGE_URL}/${PACKAGE_NAME}
17+
18+
########################################################################
19+
echo
20+
echo "Unpack weights file ${PACKAGE_NAME} ..."
21+
tar -zxvf ${PACKAGE_NAME}
22+
23+
# EfficientNet-Lite unpacks into subdirectory
24+
echo
25+
echo "Move files out of ${PACKAGE_NAME_EFFICIENTNET_LITE}/ ..."
26+
mv ${PACKAGE_NAME_EFFICIENTNET_LITE}/* ${PACKAGE_NAME_EFFICIENTNET_LITE}/..
27+
rmdir ${PACKAGE_NAME_EFFICIENTNET_LITE}
28+
echo "Keeping file ${PACKAGE_NAME_EFFICIENTNET_LITE_WITH_PRECISION}.tflite"
29+
if [ "${MODEL_EFFICIENTNET_LITE_PRECISION}" == "fp32" ]; then
30+
file_to_remove="${PACKAGE_NAME_EFFICIENTNET_LITE}-int8.tflite"
31+
echo "Removing file ${file_to_remove}"
32+
rm ${file_to_remove}
33+
else
34+
file_to_remove="${PACKAGE_NAME_EFFICIENTNET_LITE}-fp32.tflite"
35+
echo "Removing file ${file_to_remove}"
36+
rm ${file_to_remove}
37+
fi
38+
39+
########################################################################
40+
echo
41+
echo "Remove temporary files ..."
42+
rm ${PACKAGE_NAME}

0 commit comments

Comments
 (0)