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

相关新闻

最新新闻

日新闻

周新闻

月新闻