格式类型
模型训练完需要部署, 通常需要转换成其他格式,常用的格式有
- onnx
- wk
- ncnn
- caffemodel
- tensorrt
分别讲解下如何转换格式
onnx
这个比较简单, 直接用自带的export.py 就可以完成
输入python export.py
ncnn
这个比较复杂, 需要先转成onnx 的格式, 再转成ncnn 的格式, 其中有一些层可能不受支持,
需要根据具体业务需求和网络结构进行修改
caffemodel
caffe 的模型, 这个根据版本不同,需要修改的内容不同,
如1.0 版本不支持permute操作, 需要将网格生成放在后处理中实现
def forward(self, x):
z = [