Update libs/person_detector.py

libs/person_detector.py

@@ -262,8 +262,8 @@ class PersonDetectorAttrib(PersonDetector):
     def visualize(image, objs):
         bboxes = [obj['points'] for obj in objs]
         bboxes = np.array(bboxes)
-        classnames = [ACTION[obj['actions_label']] +
-                      "-" + POSE[obj['poses_label']] for obj in objs]
+        classnames = [f"ID-{i}: "+ACTION[obj['actions_label']] +
+                      "-" + POSE[obj['poses_label']] for i, obj in enumerate(objs)]
         labels = [i for i in range(len(objs))]
         # put rectangle as background of text
         # thickness=4, font_size=1 for image with size 1284 962 set thickness, font_size base on image size
@@ -274,4 +274,4 @@ class PersonDetectorAttrib(PersonDetector):
 
         image = draw_bboxes(image, bboxes, color=(
             0, 0, 255), thickness=thickness, font_size=font_size, classnames=classnames, labels=labels)
-        return image
+        return image, classnames