(base) root@davinci-mini:/home/HwHiAiUser/samples/notebooks/DeepSort# atc --model=best.onnx --framework=5 --output=yolov5s --input_format=NCHW --log=error --soc_version=Ascend310B1 --input_shape="images:1,3,640,640" ATC start working now, please wait for a moment. Exception in ...
1、首先6.0版本中有针对yolov5.pt转yolov5.onnx的export.py脚本文件,在文件中主要注意下面这个问题: 设置opset=11或者12都可以 2、转换为onnx文件后,可以通过netron地址查看一下模型的结构。 Netronnetron.app/ yolov5s的模型结构中,网络将【1,3,80,80,85】、【1,3,40,40,85】、【1,3,20,20,85】...