We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 91e9a5e commit 1797a1fCopy full SHA for 1797a1f
motrackers/detectors/yolo.py
@@ -26,7 +26,7 @@ def __init__(self, weights_path, configfile_path, labels_path, confidence_thresh
26
if cv.__version__ >= '4.6.0':
27
self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()]
28
else:
29
- self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()]
+ self.layer_names = [layer_names[i[0] - 1] for i in self.net.getUnconnectedOutLayers()]
30
31
self.scale_factor = 1/255.0
32
self.image_size = (416, 416)
0 commit comments