Tracker giving unexpected error on colab

[tanmy-sharma]

I installed boxmot on colab using pip install boxmot. I was using L4 GPU on colab pro. Please let me know how to fix this issue.

[mikel-brostrom]

I updated the argument names in README. I made some changes that were not reflected there

[tanmy-sharma]

Thank you so much for clarifying this.

I wanted to know one more thing. The following is the code to save a video using the tracker, and opencv. Now, how to extract the ids of the different cars passing alongside their coordinates. The command tracker.plot_results(im, show_trajectories=True) plots trajectories. I want the id and the coordinates of the trajectory plotted. Can you please tell me if there is a way to do that?

cap = cv2.VideoCapture('input.mp4')

frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = cap.get(cv2.CAP_PROP_FPS)

output_file = 'output.mp4'
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
out = cv2.VideoWriter(output_file, fourcc, fps, (frame_width, frame_height))

while True:
    ret, im = cap.read()
    if not ret:
        print("Reached the end of the video.")
        break

    results = model.predict(im, verbose = False)[0]
    boxes = results.boxes.xyxy.cpu().numpy()  # Convert to numpy array (x_min, y_min, x_max, y_max)
    confs = results.boxes.conf.cpu().numpy()  # Get confidence scores
    classes = results.boxes.cls.cpu().numpy()

    if len(boxes) > 0:
        dets = np.hstack([boxes, confs[:, np.newaxis], classes[:, np.newaxis]])
        tracker.update(dets, im)
    else:
        dets = np.empty((0, 6))
        tracker.update(dets, im)

    tracker.plot_results(im, show_trajectories=True)
    out.write(im)
    break

cap.release()
out.release()
cv2.destroyAllWindows()