obendidi / Tracking-with-darkflow

Real-time people Multitracker using YOLO v2 and deep_sort with tensorflow
GNU General Public License v3.0
524 stars 176 forks source link

GPU memory usage 9G #99

Open zhujiagang opened 4 years ago

zhujiagang commented 4 years ago

Hi, thanks for your repo! When I run python run.py (input is a video) on a 4 GPU machine, the code will take up 9872Mb per GPU. However, when I try to downsize the input by adjusting width=608, height=608 (to 160 for example) in darkflow/cfg/yolo.cfg, the memory usage doesn't change at all. I also try to comment the tracking code, the memory usage doesn't change. I think the memory usage is too large for detection.