fix thickness of font and bbox

Author: JackWoo0831

tracker/track_yolov5.py

@@ -249,11 +249,11 @@ def plot_img(img, frame_id, results, save_dir):
         # convert tlwh to tlbr
         tlbr = tuple([int(tlwh[0]), int(tlwh[1]), int(tlwh[0] + tlwh[2]), int(tlwh[1] + tlwh[3])])
         # draw a rect
-        cv2.rectangle(img_, tlbr[:2], tlbr[2:], get_color(id), thickness=1, )
+        cv2.rectangle(img_, tlbr[:2], tlbr[2:], get_color(id), thickness=3, )
         # note the id and cls
         text = f'{CATEGORY_DICT[cls]}-{id}'
         cv2.putText(img_, text, (tlbr[0], tlbr[1]), fontFace=cv2.FONT_HERSHEY_PLAIN, fontScale=1, 
-                        color=(255, 164, 0), thickness=1)
+                        color=(255, 164, 0), thickness=2)
 
     cv2.imwrite(os.path.join(save_dir, f'{frame_id:05d}.jpg'), img_)
 
@@ -337,4 +337,4 @@ def get_color(idx):
     # for debug
     # evaluate(sorted(os.listdir('./tracker/results/deepmot_17_08_02_38')), 
     #             sorted(os.listdir('./tracker/results/deepmot_17_08_02_38')), data_type='visdrone', result_folder='deepmot_17_08_02_38')  
-    main(opts)
+    main(opts)