theAIGuysCode / yolov4-deepsort

Object tracking implemented with YOLOv4, DeepSort, and TensorFlow.
GNU General Public License v3.0
1.32k stars 751 forks source link

deepsort is consuming huge amount of cpu #46

Open rajavi-cis opened 3 years ago

rajavi-cis commented 3 years ago

while running detection over a video, I see that my entire CPU memory is being used. I'm not able to run it on multiple threads as it leads to slowness. Did anyone face this issue ? Any help would be appreciated

allenwu5 commented 3 years ago

I have a similar issue too. And I realized GPU didn't been used because of some CUDA related errors been suppressed by:

 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 

https://github.com/theAIGuysCode/yolov4-deepsort/blob/9e745bfb3ea5e7c7505cb11a8e8654f5b1319ad9/object_tracker.py#L2-L8

Maybe you can comment os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' out to see what happened. And makesure len(physical_devices) > 0 is True.