拓冰建站拓冰建站
首页 / 资讯中心 / 正文

OpenCV DNN 加载不同框架模型

OpenCV DNN 加载不同框架模型OpenCV DNN 支持 Caffe、TensorFlow、ONNX、Torch (PyTorch)、Darknet (YOLO),不同框架对应不同加载 API。核心区分:Caffe:prototxt(结构)+caffemodel(权重)两个文件TensorFlow:.pb冻结图(结构 + 权重合一)ONNX:.onnx,跨框架通用,一个文件(PyTorch/TensorFlow 导出)Darknet(YOLO):.cfg结构 +.weights权重PyTorch 原生.pth/.pt:DNN不能直接读 pth,需要先导出为 ONNX。1、加载 Caffe 模型cpp// 方式1 readNetFromCaffe(结构文件,权重文件) Net net = readNetFromCaffe("deploy.prototxt", "model.caffemodel"); // 方式2 通用readNet,自动识别caffe Net net = readNet("model.caffemodel", "deploy.prototxt");参数 1:权重.caffemodel;参数 2:网络描述.prototxt两个文件必须配对。2、加载 TensorFlow pb 冻结模型cpp// readNetFromTen
分享:

看完干货,该让你的企业上线了

免费需求沟通 · 48 小时内出具建站方案 · 河南本地可上门