diff --git a/ACL_PyTorch/built-in/cv/Yolov5_for_Pytorch/README.md b/ACL_PyTorch/built-in/cv/Yolov5_for_Pytorch/README.md index b9278858d46c3299bd5629cdf27eb68089312097..b9f882a56ffa34c15bc651d65b41ca0eb0451f18 100644 --- a/ACL_PyTorch/built-in/cv/Yolov5_for_Pytorch/README.md +++ b/ACL_PyTorch/built-in/cv/Yolov5_for_Pytorch/README.md @@ -111,7 +111,7 @@ python3.7 models/export.py --weights=./yolov5s.pt --img-size=640 --batch-size 1 python3.7 export.py --weights=./yolov5s.pt --imgsz=640 --batch-size=1 --opset=11 # 用于v6.0 ``` -(2)对导出的onnx模型使用onnx-simplifer工具进行简化 +(2)对导出的onnx模型使用onnx-simplifier工具进行简化 ```shell python3.7 -m onnxsim --skip-optimization yolov5s.onnx yolov5s_sim.onnx @@ -158,8 +158,8 @@ python3.7 modify_model.py --model=yolov5s_sim_amct.onnx --conf-thres=0.4 --iou-t ### 3.5 利用ATC工具转换为om模型 ```shell -bash atc.sh yolov5s_sim_t.onnx yolov5s_sim_t 1 Ascend310 # 量化模型 -bash atc.sh yolov5s_sim_amct_t.onnx yolov5s_sim_t 1 Ascend310 # 非量化模型 +bash atc.sh yolov5s_sim_t.onnx yolov5s_sim_t 1 Ascend310 # 非量化模型 +bash atc.sh yolov5s_sim_amct_t.onnx yolov5s_sim_t 1 Ascend310 # 量化模型 ``` atc转模型参数说明:待转换onnx模型、转换后om模型、batchsize、运行平台(Ascend310/Ascend710) @@ -195,4 +195,4 @@ python3.7 map_calculate.py --ground_truth_json=./instances_val2017.json --detect ``` 参数说明: --ground_truth_json: 标杆标注文件 ---detection_results_json: om模型推理的结果文件 \ No newline at end of file +--detection_results_json: om模型推理的结果文件 diff --git a/ACL_PyTorch/built-in/cv/Yolov5_for_Pytorch/img.png b/ACL_PyTorch/built-in/cv/Yolov5_for_Pytorch/img.png new file mode 100644 index 0000000000000000000000000000000000000000..752d68e6af32a0e14dd71d2e4ab687bc214d8d05 Binary files /dev/null and b/ACL_PyTorch/built-in/cv/Yolov5_for_Pytorch/img.png differ