基于昇腾NPU的UNet C++部署

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