Skip to content

Commit 7eb4f8a

Browse files
committed
modify track_demo.py
1 parent de7a243 commit 7eb4f8a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tracker/track_demo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ def plot_img(img, frame_id, results, save_dir):
249249
# draw a rect
250250
cv2.rectangle(img_, tlbr[:2], tlbr[2:], get_color(id), thickness=3, )
251251
# note the id and cls
252-
text = f'car-{id}'
252+
text = f'{CATEGORY_DICT[cls]}-{id}'
253253
cv2.putText(img_, text, (tlbr[0], tlbr[1]), fontFace=cv2.FONT_HERSHEY_PLAIN, fontScale=1,
254254
color=(255, 164, 0), thickness=2)
255255

0 commit comments

Comments
 (0)