deep_sort icon indicating copy to clipboard operation
deep_sort copied to clipboard

TensorFlow 2.x with deep_sort

Open fdominik opened this issue 4 years ago • 1 comments

In case somebody needs to use TF 2.x and cant get deep_sort working, we migrated one script to be working with TF 2.x (there are still other parts of code we havent changed yet). https://github.com/mk-michal/deep_sort

The old code is using tf.Session() and Graph data, so we needed to change the structure. However as PR are not merged in this repo, we dont intend to do a PR until somebody will start maintaing this repo.

fdominik avatar Feb 03 '21 12:02 fdominik

Hello, I also needed to use tf2 and I used your repo but you didn't update freeze_model.py for tf2. Without it I'm not able to run generate_detection.py with tf2. I used tf_upgrade_v2 to convert freeze_model.py to tf2. After that and and few changes in generate_detection.py (like delate 'net/' from 97 and 99 line) I was able to run your repo and it works ok but it takes a lot of memory on my gpu (about 24GB). With tf1 model it was 160MB. Can you tell me how you converted model from original repo to run it with your 'generate_detection.py' script? Thanks

Olkaaaaaa12 avatar Feb 23 '22 09:02 Olkaaaaaa12