【pytorch】ncnn

def onnx2ncnn(self):
    assert os.path.isfile(simplified_onnx_path)

    os.system('onnx2ncnn {} {} {}'.format(simplified_onnx_path, param_path, bin_path))
    print('\n param has been save to {}'.format(param_path))
    print(' bin   has been save to {}\n'.format(bin_path))

    os.system('ncnnoptimize {} {} {} {} fp32'.format(param_path, bin_path, param_path, bin_path))
    print('\n param has been ncnnoptimized')
    print(' bin   has been ncnnoptimized\n')

pip install onnx-simplifier==0.2.4 -i https://pypi.tuna.tsinghua.edu.cn/simple 

python -m onnxsim .\2task.aiyou.0.0.onnx 2task.0.0.onnx

ncnnoptimize A.param A.bin A.param A.bin fp32