@@ -247,11 +247,11 @@ def plot_img(img, frame_id, results, save_dir):
247
247
# convert tlwh to tlbr
248
248
tlbr = tuple ([int (tlwh [0 ]), int (tlwh [1 ]), int (tlwh [0 ] + tlwh [2 ]), int (tlwh [1 ] + tlwh [3 ])])
249
249
# 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 , )
251
251
# note the id and cls
252
252
text = f'{ CATEGORY_DICT [cls ]} -{ id } '
253
253
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 )
255
255
256
256
cv2 .imwrite (filename = os .path .join (save_dir , f'{ frame_id :05d} .jpg' ), img = img_ )
257
257
@@ -336,4 +336,4 @@ def get_color(idx):
336
336
# for debug
337
337
# evaluate(sorted(os.listdir('./tracker/results/deepmot_17_08_02_38')),
338
338
# 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