Skip to content

Commit 53d3b39

Browse files
authored
Merge pull request #1 from devmdave/devmdave-patch-1
Updated Yolo.py for the index error issue.
2 parents 1797a1f + 60bb1c0 commit 53d3b39

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

motrackers/detectors/yolo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def __init__(self, weights_path, configfile_path, labels_path, confidence_thresh
2626
if cv.__version__ >= '4.6.0':
2727
self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()]
2828
else:
29-
self.layer_names = [layer_names[i[0] - 1] for i in self.net.getUnconnectedOutLayers()]
29+
self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()]
3030

3131
self.scale_factor = 1/255.0
3232
self.image_size = (416, 416)

0 commit comments

Comments
 (0)