@@ -247,11 +247,11 @@ def plot_img(img, frame_id, results, save_dir):
247247 # convert tlwh to tlbr
248248 tlbr = tuple ([int (tlwh [0 ]), int (tlwh [1 ]), int (tlwh [0 ] + tlwh [2 ]), int (tlwh [1 ] + tlwh [3 ])])
249249 # draw a rect
250- cv2 .rectangle (img_ , tlbr [:2 ], tlbr [2 :], get_color (id ), thickness = 1 , )
250+ cv2 .rectangle (img_ , tlbr [:2 ], tlbr [2 :], get_color (id ), thickness = 3 , )
251251 # note the id and cls
252252 text = f'{ CATEGORY_DICT [cls ]} -{ id } '
253253 cv2 .putText (img_ , text , (tlbr [0 ], tlbr [1 ]), fontFace = cv2 .FONT_HERSHEY_PLAIN , fontScale = 1 ,
254- color = (255 , 164 , 0 ), thickness = 1 )
254+ color = (255 , 164 , 0 ), thickness = 2 )
255255
256256 cv2 .imwrite (filename = os .path .join (save_dir , f'{ frame_id :05d} .jpg' ), img = img_ )
257257
@@ -336,4 +336,4 @@ def get_color(idx):
336336 # for debug
337337 # evaluate(sorted(os.listdir('./tracker/results/deepmot_17_08_02_38')),
338338 # sorted(os.listdir('./tracker/results/deepmot_17_08_02_38')), data_type='visdrone', result_folder='deepmot_17_08_02_38')
339- main (opts )
339+ main (opts )
0 commit comments