-
Notifications
You must be signed in to change notification settings - Fork 150
/
dnnc_miniGoogleNet.sh
41 lines (26 loc) · 1.26 KB
/
dnnc_miniGoogleNet.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#!/bin/bash
DNNDK_ROOT=$HOME/ML/DNNDK/tools
net=miniGoogleNet
work_dir=$HOME/ML/cifar10/deephi/miniGoogleNet/quantiz
model_dir=${work_dir}/decent_output
output_dir=${work_dir}/dnnc_output
echo "Compiling network: ${net}"
$DNNDK_ROOT/dnnc --prototxt=${model_dir}/deploy.prototxt \
--caffemodel=${model_dir}/deploy.caffemodel \
--output_dir=${output_dir} \
--net_name=${net} \
--dpu=4096FA \
--cpu_arch=arm64 \
--mode=debug \
--save_kernel
echo " copying dpu elf file into /../zcu102/baseline/model/arm64_4096 "
cp ${output_dir}/dpu_${net}\_*.elf ${output_dir}/../../zcu102/baseline/model/arm64_4096
echo " copying the test images to be used by the ZCU102"
cp -r $HOME/ML/cifar10/input/cifar10_jpg/test ${output_dir}/../../zcu102/test_images
mv ${net}_kernel*.* ${work_dir}/dnnc_output
# The following lines are for documentation only. Leave them commented
# now you can create the tar archive that you will then scp/ssh copy from HOST PC to TARGET board
## cd ~/ML/cifar10/deephi/miniVggNet/zcu102/baseline
## ln -s ../test_images ./test_images
## cd ..
## tar -xvf baseline.tar ./baseline