53 lines
1.0 KiB
Bash
Executable File
53 lines
1.0 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
show_usage() {
|
|
echo
|
|
echo "Usage: $0 demo/benchmark"
|
|
echo
|
|
}
|
|
|
|
init_workspace() {
|
|
rm -rf occlum_workspace
|
|
mkdir occlum_workspace
|
|
cd occlum_workspace
|
|
occlum init
|
|
}
|
|
|
|
build_occlum() {
|
|
cp ../tensorflow_src/tensorflow/lite/tools/make/gen/linux_x86_64/bin/* image/bin
|
|
cp /usr/local/occlum/x86_64-linux-musl/lib/libz.so.1 image/lib
|
|
cp -r ../models image
|
|
cp -r ../tensorflow_src/tensorflow/lite/examples/label_image/testdata image
|
|
occlum build
|
|
}
|
|
|
|
run_demo() {
|
|
init_workspace
|
|
build_occlum
|
|
occlum run /bin/label_image \
|
|
--tflite_model ./models/mobilenet_v1_1.0_224.tflite \
|
|
--labels ./models/labels.txt \
|
|
--image ./testdata/grace_hopper.bmp
|
|
}
|
|
|
|
run_benchmark() {
|
|
init_workspace
|
|
build_occlum
|
|
occlum run /bin/benchmark_model \
|
|
--graph=./models/mobilenet_v1_1.0_224.tflite \
|
|
--warmup_runs=5
|
|
}
|
|
|
|
bin=$1
|
|
case "$bin" in
|
|
demo)
|
|
run_demo
|
|
;;
|
|
benchmark)
|
|
run_benchmark
|
|
;;
|
|
*)
|
|
show_usage
|
|
esac
|