加载 YOLO:标量变量的无效索引
Loading YOLO: invalid index to scalar variable
收到 IndexError 错误:yolo_layers 行上的标量变量索引无效。
network = cv2.dnn.readNetFromDarknet('yolov3.cfg', 'yolov3.weights')
layers = network.getLayerNames()
yolo_layers = [layers[i[0] - 1] for i in network.getUnconnectedOutLayers()]
此代码无法在我的 Jupyter 笔记本上运行,但 运行 可以在 google collab 上运行。不知道为什么。可能是我的 python 版本?
可能是cv2版本不同造成的。支持 CUDA 的 cv2 模块版本将在调用 network.getUnconnectedOutLayers() 时为您提供一个二维数组。但是,不支持CUDA的版本会给出一维数组。
您可以尝试去掉关闭索引 0 的括号。
收到 IndexError 错误:yolo_layers 行上的标量变量索引无效。
network = cv2.dnn.readNetFromDarknet('yolov3.cfg', 'yolov3.weights')
layers = network.getLayerNames()
yolo_layers = [layers[i[0] - 1] for i in network.getUnconnectedOutLayers()]
此代码无法在我的 Jupyter 笔记本上运行,但 运行 可以在 google collab 上运行。不知道为什么。可能是我的 python 版本?
可能是cv2版本不同造成的。支持 CUDA 的 cv2 模块版本将在调用 network.getUnconnectedOutLayers() 时为您提供一个二维数组。但是,不支持CUDA的版本会给出一维数组。
您可以尝试去掉关闭索引 0 的括号。