我几乎搜索了互联网上的每个地方,但找不到以下功能的工作和输出。特别是他们在YOLO
算法中所做的。
getLayerNames()
getUnconnectedOutLayers()
Run Code Online (Sandbox Code Playgroud)
代码如下:
import cv2
import numpy as np
import time
#Loading Yolo
net = cv2.dnn.readNet("yolov3.weights", "yolov3.cfg")
classes = []
with open("coco.names", "r") as f:
classes = [line.strip() for line in f.readlines()]
layer_names = net.getLayerNames()
outputlayers=[layer_names[i[0] - 1] for i in net.getUnconnectedOutLayers()]
Run Code Online (Sandbox Code Playgroud)
如图所示,YOLOv3 有 3 个输出层(82、94 和 106)。
getLayerNames():获取网络所有层的名称。
getUnconnectedOutLayers():获取输出层的索引。
这两个函数用于获取输出层 (82,94,106)。为简单起见,我更喜欢使用以下代码:
import cv2
import numpy as np
import time
#Loading Yolo
net = cv2.dnn.readNet("yolov3.weights", "yolov3.cfg")
classes = []
with open("coco.names", "r") as f:
classes = [line.strip() for line in f.readlines()]
outputlayers = net.getUnconnectedOutLayersNames()
Run Code Online (Sandbox Code Playgroud)
图像参考(外部链接)
归档时间: |
|
查看次数: |
3650 次 |
最近记录: |