search

aditya-vora / FCHD-Fully-Convolutional-Head-Detector

Code for FCHD - A fast and accurate head detector
Other
647 stars 182 forks source link

RuntimeError: CUDA error: out of memory #51

Open tingnini opened 5 years ago

tingnini commented 5 years ago

运行demo 的时候： python head_detection_demo.py --img_path mov_021_178867.jpeg --model_path '/FCHD-Fully-Convolutional-Head-Detector/checkpoints/head_detector09191214_0.6141253457131631' 报错：RuntimeError: CUDA error: out of memory