How do I train the DeepSORT tracker for a custom class?

This page summarizes the projects mentioned and recommended in the original post on /r/computervision

Our great sponsors
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • WorkOS - The modern identity platform for B2B SaaS
  • SaaSHub - Software Alternatives and Reviews
  • yolov4-deepsort

    Object tracking implemented with YOLOv4, DeepSort, and TensorFlow.

    I am trying to track objects in a sequence of images in order to count them. I was looking around for robust trackers since in my case, the camera moves with respect to the object. I found the DeepSORT tracker online and it seems like the solution to my problem. However, I am not sure of how I could train it for my own custom classes. I am currently looking at this repository and it seems to almost do the things I want, except for the counting part. Can anyone explain to me how I can train the DeepSORT tracker for my own classes? I am already training a YOLOv4 model on these custom classes. As a result, I have collected a labelled dataset for the training and validation purposes, and if I have to use images for the training.

  • deep_sort

    Simple Online Realtime Tracking with a Deep Association Metric

    I was wondering if I could use the same annotated data(in YOLO format) for the training of the tracker as well. I took a look at the original repo for DeepSORT, and it does mention the training using cosine metric learning, but I could not seem to understand how to replicate that for my own dataset(they show us how to do it for the MARS and Market1501 datasets).

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts