Python图像识别训练闭环：从数据标注到YOLOv8部署
2026/10/13 12:06:15
文章目录
- PT转ONNX
- ONNX转OM
- 关键代码
PT转ONNX
从https://github.com/milesial/Pytorch-UNet中下载pt文件
修改predict.py文件，mode = “export_onnx”
python predict.pyONNX转OM
atc --framework=5--model=models.onnx --input_format=NCHW --input_shape="images:1,3,512,512"--output_type