用户您好,请详细描述您所遇到的问题,这会帮助我们快速定位问题~
1.芯片型号:J5
2.天工开物开发包OpenExplorer版本:j5_open_explorer_v1.1.57
3.问题定位:例如模型转换
4.问题具体描述:
我下载了v2.0的yolov5x.pt,然后生成了best.pt,之后尝试转换onnx格式
发现leakyrelu 被替换为sigmoid,而且不容易更改输出节点按照1.3、1.4不太明白
请问是否可以提供详细操作方式
1.3 after download, convert pt file into ONNX file using the `https://github.com/ultralytics/yolov5/blob/v2.0/models/export.py\` script
1.4 As for YOLOv5 model, in terms of model structure, we modified some output nodes. As currently OpenExplorer XJ3 Toolchain doesn’t support 5-dimensional Reshape, we deleted it in the prototxt and moved it to post-process. Meanwhile, we’ve aslo added a transpose operator to enable the node to dump NHWC. This is because in Horizon’s ASIC, BPU hardware runs NHWC layout, therefore, BPU can directly dump results after some modifications, rather than introducing additional transpose in quantized model. For more details please refer to the text and table in the Benchmark section.
参考