Skip to content

Commit 3aa4de1

Browse files
authored
Merge pull request adipandas#52 from partheee/patch-1
Update yolo.py
2 parents 18565a2 + b8ac59a commit 3aa4de1

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
@@ -23,7 +23,7 @@ def __init__(self, weights_path, configfile_path, labels_path, confidence_thresh
2323
object_names = load_labelsjson(labels_path)
2424

2525
layer_names = self.net.getLayerNames()
26-
if cv2.__version__ == '4.6.0':
26+
if cv.__version__ >= '4.6.0':
2727
self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()]
2828
else:
2929
self.layer_names = [layer_names[i[0] - 1] for i in self.net.getUnconnectedOutLayers()]

0 commit comments

Comments
 (0)