opencv用dnn.readNet加载caffe/torch/darknet/tensorflow的模型和权重

3 篇文章 0 订阅
2 篇文章 0 订阅

cv.dnn.readNet()

官方文档:
https://docs.opencv.org/3.4/d6/d0f/group__dnn.html#ga3b34fe7a29494a6a4295c169a7d32422

Python使用方法举例(Examples for Python):

如使用Darknet的预训练模型:

weights = "yolov2.weights"
config_file = "yolov2.cfg"
# read pre-trained model and config file
net = cv2.dnn.readNet(weights, config_file)

如使用Caffe的预训练模型:

path_to_prototxt = "deploy.prototxt"
path_to_caffemodel = "hed_pretrained_bsds.caffemodel"
net = cv.dnn.readNet(path_to_prototxt, path_to_caffemodel)

C++使用方法举例(Examples for C++):

samples/dnn/classification.cpp;
samples/dnn/object_detection.cpp;
samples/dnn/openpose.cpp;
samples/dnn/segmentation.cpp;
samples/dnn/text_detection.cpp.

可使用的模型(Available Model):

Caffe, TensorFlow, Torch, Darknet, DLDT, ONNX

模型权重文件支持的文件格式:

  • *.caffemodel (Caffe, http://caffe.berkeleyvision.org/)
  • *.pb (TensorFlow, https://www.tensorflow.org/)
  • *.t7 | *.net (Torch, http://torch.ch/)
  • *.weights (Darknet, https://pjreddie.com/darknet/)
  • *.bin (DLDT, https://software.intel.com/openvino-toolkit)
  • *.onnx (ONNX, https://onnx.ai/)

模型配置文件支持的文件格式:

*.prototxt (Caffe, http://caffe.berkeleyvision.org/)
*.pbtxt (TensorFlow, https://www.tensorflow.org/)
*.cfg (Darknet, https://pjreddie.com/darknet/)
*.xml (DLDT, https://software.intel.com/openvino-toolkit)

可见不支持pytorch的*.pth和keras等的*.h5
在这里插入图片描述

cv.dnn 的其它加载模型的方法有:

  • readNetFromCaffe()
  • readNetFromDarknet()
  • readNetFromModelOptimizer()
  • readNetFromONNX()
  • readNetFromTensorflow()
  • readNetFromTorch()

支持的文件格式同readNet()中相应的模型

  • 5
    点赞
  • 53
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值