企业官网建设流程全解析

PT转ONNX

从https://github.com/milesial/Pytorch-UNet中下载pt文件

修改predict.py文件，mode = “export_onnx”

python predict.py

ONNX转OM

atc --framework=5--model=models.onnx --input_format=NCHW --input_shape="images:1,3,512,512"--output_type