From e413121708d7de9a4aa0c56dd48aad1e3827f4ad Mon Sep 17 00:00:00 2001 From: kasyanikita <71206801+kasyanikita@users.noreply.github.com> Date: Wed, 23 Aug 2023 13:44:50 +0300 Subject: [PATCH] Update yolo.py Fix bug with cv2 --- motrackers/detectors/yolo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/motrackers/detectors/yolo.py b/motrackers/detectors/yolo.py index d6fd24d..894e911 100644 --- a/motrackers/detectors/yolo.py +++ b/motrackers/detectors/yolo.py @@ -23,10 +23,10 @@ def __init__(self, weights_path, configfile_path, labels_path, confidence_thresh object_names = load_labelsjson(labels_path) layer_names = self.net.getLayerNames() - if cv2.__version__ == '4.6.0': + if cv.__version__ == '4.6.0': self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()] else: - self.layer_names = [layer_names[i[0] - 1] for i in self.net.getUnconnectedOutLayers()] + self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()] self.scale_factor = 1/255.0 self.image_size = (416, 416)