Real-time Multi-person tracker using YOLO v3 and deep_sort with tensorflow
GNU General Public License v3.0
1.66k
stars
592
forks
source link
i want to set the gpu fraction,but it failed, gpu is always fully occupied #170
Open
zfs1993 opened 4 years ago
i add these code to set the gpu fraction, the yolo part (yolo.py) class YOLO(object): def init(self): self.model_path = 'model_data/yolo.h5'
self.model_path = 'model_data/yolo_tiny.h5'
the features part(tools/generate_detections.py) class ImageEncoder(object):
but it failed ,is there anyone who meet the same questions?