supervision icon indicating copy to clipboard operation
supervision copied to clipboard

Include DEEP SORT TRACKING

Open yeongnamtan opened this issue 2 years ago • 3 comments

Search before asking

  • [X] I have searched the Supervision issues and found no similar feature requests.

Description

Would you consider including DEEP SORT Tracker in Supervision ?

Use case

No response

Additional

No response

Are you willing to submit a PR?

  • [ ] Yes I'd like to help by submitting a PR!

yeongnamtan avatar Nov 14 '23 06:11 yeongnamtan

Hi, @yeongnamtan! 👋🏻 Correct me if I'm wrong, but DeepSort requires loading the model in PyTorch or TensorFlow. We are trying to make Supervision not require installation of such heavy dependencies.

SkalskiP avatar Nov 15 '23 14:11 SkalskiP

@SkalskiP I did a comparison between ByteTrack (using LINE COUNTER), and DeepSort for the same source video. DeepSort performed much better in terms of counting. Picture1

For ByteTrack, my settings as follow: byte_tracker = sv.ByteTrack(track_thresh=0.3, track_buffer=60, match_thresh=0.9, frame_rate=30)

For DeepSort, confidence level same at 30% DEEPSORT: MODEL_TYPE: "osnet_x_25" MAX_DIST: 0.1 # The matching threshold. Samples with larger distance are considered an invalid match MAX_IOU_DISTANCE: 0.7 # Gating threshold. Associations with cost larger than this value are disregarded. MAX_AGE: 30 # Maximum number of missed misses before a track is deleted N_INIT: 3 # Number of frames that a track remains in initialization phase NN_BUDGET: 100 # Maximum size of the appearance descriptors gallery

yeongnamtan avatar Nov 16 '23 07:11 yeongnamtan