Skip to content

Commit 448d8d1

Browse files
authored
fix thickness of font and bbox
1 parent fed6278 commit 448d8d1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tracker/track.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)