theAIGuysCode / yolov4-deepsort

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

Memory Issue #106

Open sekisek opened 2 years ago

sekisek commented 2 years ago

Everything is working for me, but it takes more than 8GB of GPU. (nvidia-smi) I encountered the same thing with Darknet and was able to reduce the memory use by reducing the resolution (width/height) of the network in the configuration file, is there a way I can do it here with TensorFlow?

if not is there a way to unload the network from memory in python and reload it again when needed? or any other solution?

thanks!