Update detect_strongsort.py

detect_strongsort.py

@@ -44,7 +44,7 @@ VID_FORMATS = 'asf', 'avi', 'gif', 'm4v', 'mkv', 'mov', 'mp4', 'mpeg', 'mpg', 't
 def plot_one_box(x, img, color=None, label=None, line_thickness=3):
     # Plots one bounding box on image img
     tl = line_thickness or round(0.002 * (img.shape[0] + img.shape[1]) / 2) + 1  # line/font thickness
-    color = color or [random.randint(0, 255) for _ in range(3)]
+    #color = color or [random.randint(0, 255) for _ in range(3)]
     c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3]))
     cv2.rectangle(img, c1, c2, color, thickness=tl, lineType=cv2.LINE_AA)
     if label:
@@ -157,7 +157,7 @@ def run_strongsort(
         strongsort_list[i].model.warmup()
     outputs = [None] * bs
     
-    colors = [[random.randint(0, 255) for _ in range(3)] for _ in names]
+    colors = [[0, 0, 255], [0,148,255], [0, 255, 10], [250, 247, 0], [235,0,255]]
 
     # Run tracking
     model.warmup(imgsz=(1 if pt or model.triton else bs, 3, *imgsz))  # warmup