121 OpenCV DNN 获取导入模型各层信息
代码
import cv2 as cv
import numpy as np
bin_model = "../models/googlenet/bvlc_googlenet.caffemodel"
protxt = "../models/googlenet/bvlc_googlenet.prototxt"
# Load names of classes
classes = None
with open("classification_classes_ILSVRC2012.txt", 'rt') as f:
classes = f.read().rstrip('\n').split('\n')
# load CNN model
net = cv.dnn.readNet(bin_model, protxt)
# 获取各层信息
layer_names = net.getLayerNames()
print(layer_names)
for name in layer_names:
id = net.getLayerId(name)
layer = net.getLayer(id)
print("layer id : %d, type : %s, name: %s"%(id, layer.type, layer.name))
print("successfully loaded model...")
实验结果
解释
模型支持1000个类别的图像分类,OpenCV DNN模块支持下面框架的预训练模型的前馈网络(预测图)使用
- Caffe
- Tensorflow
- Torch
- DLDT
- Darknet
同时还支持自定义层解析、非最大抑制操作、获取各层的信息等。OpenCV加载模型的通用API为
retval = cv.dnn.readNet(model[, config[, framework]])
model二进制训练好的网络权重文件,可能来自支持的网络框架,扩展名为如下:
*.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)
config针对模型二进制的描述文件,不同的框架配置文件有不同扩展名
*.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)
framework显示声明参数,说明模型使用哪个框架训练出来的
所有内容均来源于贾志刚老师的知识星球——OpenCV研习社,本文为个人整理学习,已获得贾老师授权,有兴趣、有能力的可以加入贾老师的知识星球进行深入学习。