search

deepjavalibrary / djl

An Engine-Agnostic Deep Learning Framework in Java
https://djl.ai
Apache License 2.0
4.06k stars 648 forks source link

Multi-target tracking #1219

Open chengpengvb opened 2 years ago

chengpengvb commented 2 years ago

At present, I can get the recognition result of each frame by grabbing the rtsp stream. Now I have a problem. I have not found a suitable target tracking algorithm for java. Something like deepsort. Someone encounters similar problems or does Have a similar algorithm work, thanks

frankfliu commented 2 years ago

@chengpengvb you can try to port deepsort into Java using DJL.

lchunleo commented 2 years ago

@frankfliu can i check for the porting, what should the approach be? is it line by line code porting? thanks

frankfliu commented 2 years ago

@chengpengvb There is a pretrained deepsort model in PyTorch: https://github.com/ZQPei/deep_sort_pytorch

You can trace the model and convert to torchscript model. Then you can load it in DJL.

However you still need implement Tracker, Kalman filter, feature extraction, etc.