Merge pull request adipandas#40 from partheee/patch-1

Update in yolo.py

Author: adipandas

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)