From 4c74affa8e78fd6e56958437be052001fec002eb Mon Sep 17 00:00:00 2001 From: Parthiban Marimuthu <66585214+ParthibanMarimuthu@users.noreply.github.com> Date: Fri, 25 Nov 2022 18:19:52 +0530 Subject: [PATCH] Update in yolo.py Fixing Bug in detector_YOLOv3.py #37 --- motrackers/detectors/yolo.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/motrackers/detectors/yolo.py b/motrackers/detectors/yolo.py index 026a556..d6fd24d 100644 --- a/motrackers/detectors/yolo.py +++ b/motrackers/detectors/yolo.py @@ -23,7 +23,10 @@ def __init__(self, weights_path, configfile_path, labels_path, confidence_thresh object_names = load_labelsjson(labels_path) layer_names = self.net.getLayerNames() - self.layer_names = [layer_names[i[0] - 1] for i in self.net.getUnconnectedOutLayers()] + if cv2.__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.scale_factor = 1/255.0 self.image_size = (416, 416)