zishuo/ascend_community_projects
1
0
Fork 0
mirror of https://github.com/Ascend/ascend_community_projects.git synced 2025-09-27 04:05:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

update HelmetIdentification_V2/model/act-env.sh.

Browse Source 
Signed-off-by: 叶生 <2930043128@qq.com>
This commit is contained in:
叶生
2022-12-12 07:15:55 +00:00
committed by Gitee
parent 12e0c03c03
commit 405af95e70
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
HelmetIdentification_V2/model/act-env.sh
View File

@@ -0,0 +1,14 @@
#!/bin/bash
# This is used to convert onnx model file to .om model file.
export install_path=/usr/local/Ascend/ascend-toolkit/latest
export PATH=/usr/local/python3.9.2/bin:${install_path}/arm64-linux/atc/ccec_compiler/bin:${install_path}/arm64-linux/atc/bin:$PATH
export PYTHONPATH=${install_path}/arm64-linux/atc/python/site-packages:${install_path}/arm64-linux/atc/python/site-packages/auto_tune.egg/auto_tune:${install_path}/arm64-linux/atc/python/site-packages/schedule_search.egg
export LD_LIBRARY_PATH=${install_path}/arm64-linux/atc/lib64:$LD_LIBRARY_PATH
export ASCEND_OPP_PATH=${install_path}/opp
export Home="./path/"
# Home is set to the path where the model is located
# Execute, transform YOLOv5 model.
atc --model="${Home}"/YOLOv5_s.onnx --framework=5 --output="${Home}"/YOLOv5_s --insert_op_conf=./aipp_YOLOv5.config --input_format=NCHW --log=info --soc_version=Ascend310 --input_shape="images:1,3,640,640"
# --model is the path where onnx is located. --output is the path where the output of the converted model is located