在yolo_layers行上遇到IndexError:无效的标量变量索引错误。
network = cv2.dnn.readNetFromDarknet('yolov3.cfg', 'yolov3.weights')layers = network.getLayerNames()yolo_layers = [layers[i[0] - 1] for i in network.getUnconnectedOutLayers()]
这段代码在我的Jupyter笔记本上无法运行,但在Google Colab上可以正常运行。不知道为什么。可能是我的Python版本问题?
回答:
这可能是由不同版本的cv2引起的。带CUDA支持的cv2模块版本在调用network.getUnconnectedOutLayers()
时会返回一个二维数组。然而,不带CUDA支持的版本会返回一个一维数组。
你可以尝试去掉索引0的括号。