-
Notifications
You must be signed in to change notification settings - Fork 0
/
compile-myriad2.yaml
51 lines (38 loc) · 1.13 KB
/
compile-myriad2.yaml
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
apiVersion: tekton.dev/v1beta1
kind: Task
metadata:
name: compile-myriad2
spec:
params:
- name: model-name
type: string
results:
- description: the name from the model config.json file
name: model-name
type: string
- description: the name from the unn file
name: unn
type: string
steps:
- image: quay.io/iain_keaney/cuala
name: ''
resources: {}
script: >
#!/usr/bin/env bash
cd /workspace/output
MODELNAME=$(params.model-name)
echo ${MODELNAME} | tee $(results.model-name.path)
XML=$MODELNAME.xml
UNN=$MODELNAME.unn
echo ${UNN} | tee $(results.unn.path)
source /opt/intel/openvino_2020.3.355/bin/setupvars.sh
/opt/intel/openvino_2020.3.355/deployment_tools/inference_engine/lib/intel64/myriad_compile
-m "${XML}" -VPU_MYRIAD_PLATFORM VPU_MYRIAD_2450 -VPU_NUMBER_OF_SHAVES 8
-VPU_NUMBER_OF_CMX_SLICES 8 -o "resnet_v1-50.unn"
ls -a
pwd
echo ${UNN}
exit 0
workspaces:
- description: Compiles .unn model for the Myriad 2 device.
name: output