small_wu的博客

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