记录一次关于ncnn部署经历
一、 写在前面第一次做移动端部署,经验为零,不断的学习和摸索,完成相关工作,书写只是为了和大佬交流,记录相关工作。二、模型转换1.模型转onnx这里没有好强调的,总而言之坑比较多,我用的模型里面作者重写了state_dict()导致我的模型导入一直有问题,查看了2天代码才找到,完成了onnx转换。主要代码:torch.onnx.export(model, input_size, onnx_path, verbose=True, input_names=['input'], output_nam
原创
2021-09-18 11:09:33 ·
1277 阅读 ·
2 评论