![](https://img-blog.csdnimg.cn/20190918140213434.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
模型转换
实现当前pkl、onnx、darknet和caffe等的模型转换
wyjfjs
这个作者很懒,什么都没留下…
展开
-
yolox的caffe模型推理(从pt模型转换成caffe模型后的推理不包括模型转换)
定位到yolov5_caffe目录下cd toolsvim caffe_yolov5s.cpp设置如下参数:INPUT_W(模型输入宽度)INPUT_H(模型输入高度)NUM_CLASS(模型有多少个类别,例如我训练的模型是安全帽检测,只有1类,所以设置为1,不需要加背景类)NMS_THRESH(做非极大值抑制的阈值)CONF_THRESH(类别置信度)prototxt_path(caffe模型的prototxt路径)caffemodel_path(caffe模型的caffemodel路原创 2022-01-06 10:35:32 · 1354 阅读 · 0 评论 -
pkl转onnx模型
实现模型的转换代码如下#from sklearn.externals import joblib#import joblib#from skl2onnx import convert_sklearn#from skl2onnx.common.data_types import onnxmltoolsimport torchenimport torchvisionimport torchvision.models as modelsfrom lib import Model, ClassAv原创 2022-01-05 16:07:33 · 2258 阅读 · 0 评论