Detection category

[zhangbaoj]

Hello, if I want to output the target category to the detection image, where should I modify?

[zhangbaoj]

In order to display the target category, I added the following content to line 81 of plot.py:

cv2.putText(img, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], lineType=cv2.LINE_AA)

The following content is displayed in the result of the detection image:

Does the 0 in the image represent the category, and does 0.92 represent the confidence level?

[Riser6]

Yes, you are right. Thanks for your interest in our project.

[zhangbaoj]

你们的工程是不是把'car', 'bus': , 'truck', 'train'都归为类别0了？

[Riser6]

yes

[stanny880913]

Hi, may I ask how to change 0 to class name like car, truck...? thank you