kcg2015 / Vehicle-Detection-and-Tracking

Computer vision based vehicle detection and tracking using Tensorflow Object Detection API and Kalman-filtering
533 stars 191 forks source link

Made it work with Tensorflow 2.1.0 and sklearn 0.23.1 and fixed some … #31

Open Jugurtha opened 4 years ago

Jugurtha commented 4 years ago

Made it work with TensorFlow 2.1.0 (using the ".v1." trick in the function calls) and sklearn 0.23.1 (by switching from sklearn.utils.linearassignment to scipy.optimize.linear_sum_assignment) and fixed some weird import bug in tracker and helpers. I didn't test it with a GPU because I don't have a CUDA-based one.

annanyatyagi12 commented 2 years ago

This works much better!