使用华为Atlas 2001Dk A2导入自己训练的yolov5模型,在开始执行模型后没有报错也不输出结果,始终在加载
收藏回复举报
使用华为Atlas 2001Dk A2导入自己训练的yolov5模型,在开始执行模型后没有报错也不输出结果,始终在加载
t('forum.solved') 已解决
新人帖
发表于2024-03-20 13:01:44
0 查看
 在使用华为华为Atlas 2001Dk A2中,按照https://gitee.com/ascend/samples/tree/master/inference/modelInference/sampleYOLOV7中所给的c++案例,进行环境配置后可以运行其中的模型,将代码中的模型进行修改后,训练yolov5官网模型也可以运行(同样使用该文件下的config.aipp),在自己训练的yolov5的模型中,通过其链接中所给的config.aipp进行转换,
atc --model=test.onnx --framework=5 --output=test --input_shape="images:1,3,640,640"  --soc_version=Ascend310B4  --insert_op_conf=aipp.cfg

之后进行配置编译,运行sample_build.sh和.shsample_run.sh后会显示以下界面:

(base) root@davinci-mini:/usr/local/samples/inference/modelInference/sampleYOLOV5_best/scripts# bash sample_build.sh 

[INFO] Sample preparation 

[INFO] The yolov7x.om already exists.start buiding 

-- The C compiler identification is GNU 11.3.0 

-- The CXX compiler identification is GNU 11.3.0 

-- Detecting C compiler ABI info 

-- Detecting C compiler ABI info - done 

-- Check for working C compiler: /usr/bin/cc - skipped 

-- Detecting C compile features 

-- Detecting C compile features - done 

-- Detecting CXX compiler ABI info 

-- Detecting CXX compiler ABI info - done 

-- Check for working CXX compiler: /usr/bin/g++ - skipped 

-- Detecting CXX compile features 

-- Detecting CXX compile features - done 

-- set default INC_PATH: /usr/local/Ascend/ascend-toolkit/latest 

-- set default LIB_PATH: /usr/local/Ascend/ascend-toolkit/lastest/runtime/lib64/stub 

-- set THIRDPART: /usr/local/Ascend/thirdpart/aarch64 

-- Configuring done 

-- Generating done 

-- Build files have been written to: /usr/local/samples/inference/modelInference/sampleYOLOV5_best/build/intermediates/host 

[ 50%] Building CXX object CMakeFiles/main.dir/sampleYOLOV7.cpp.o 

[100%] Linking CXX executable /usr/local/samples/inference/modelInference/sampleYOLOV5_best/out/main 

[100%] Built target main 

[INFO] Sample preparation is complete 

(base) root@davinci-mini:/usr/local/samples/inference/modelInference/sampleYOLOV5_best/scripts# bash sample_run.sh 

[INFO] The sample starts to run 

[INFO]  Acl init ok 

[INFO]  Open device 0 ok 

[INFO]  Use default context currently 

[INFO]  dvpp init resource ok 

[INFO]  Load model ../model/best_sar_7.om success 

[INFO]  Create model description success 

[INFO]  Create model(../model/best_sar_7.om) output success 

[INFO]  Init model ../model/best_sar_7.om success 

 之后就一直停在这一步,在主函数文件打完log发现停在了ret = model_.Execute(inferOutputs);//加载模型这一步

想问一下应该如何解决?

我要发帖子