Example 8: Use real-time evaluation tools#

ℹ️ Source code and input file(s) -> {pyppbox repo}/examples

#################################################################################
# Example of using pyppbox's evatools
#################################################################################

import cv2

from pyppbox.standalone import setMainModules, detectPeople, trackPeople, reidPeople
from pyppbox.utils.visualizetools import visualizePeople
from pyppbox.utils.evatools import MyEVA # Import `MyEVA`


# Set the main modules using the internal configurations of pyppbox
main_configurations = {'detector': 'YOLO_Classic', 
                       'tracker': 'SORT', 
                       'reider': 'Torchreid'} # Torchreid is used -> Line 24
setMainModules(main_yaml=main_configurations) 

# Use MyEVA
eva = MyEVA()

# The "gta.mp4.txt" is ground-truth for "gta.mp4"
gt_file = "data/gta.mp4.txt" 

# id_mode="deepid" is for "Torchreid" -> Line 15
eva.setGTByKnownGTFile(gt_file=gt_file, id_mode="deepid") 

# (Optional) MyEVA requires you to help count reid from `reidPeople()`
total_reid = 0 

input_video = "data/gta.mp4"
cap = cv2.VideoCapture(input_video)

while cap.isOpened():
    hasFrame, frame = cap.read()

    if hasFrame:

        # Detect people without visualizing
        detected_people, _ = detectPeople(frame, img_is_mat=True, visual=False)

        # Track the detected people
        tracked_people = trackPeople(frame, detected_people, img_is_mat=True)

        # Re-identify the tracked people
        reidentified_people, reid_count = reidPeople(
            frame, 
            tracked_people, 
            img_is_mat=True
        )

        # Count reid
        total_reid = total_reid + reid_count[0] + reid_count[1]

        # Visualize people in video frame with reid status `show_reid=reid_count`
        visualized_mat = visualizePeople(
            frame, 
            reidentified_people, 
            show_reid=reid_count
        )
        cv2.imshow("pyppbox: example_08_evatools.py", visualized_mat)

        # MyEVA -> Validate the `reidentified_people`
        eva.validate(reidentified_people)

        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    else:
        break
cap.release()

# MyEVA -> Get and print evaluation summary
eva.setReIDcount(total_reid)
wrong_id, missed_det, fault_det, reid_count, total_det, score = eva.getSummary(print_summary=True)