search

chuanqi305 / MobileNet-SSD

Caffe implementation of Google MobileNet SSD detection network, with pretrained weights on VOC0712 and mAP=0.727.
MIT License
2k stars 1.18k forks source link

model time #163

Open taxuezcy opened 4 years ago

taxuezcy commented 4 years ago

@chuanqi305 ,Hi, i use my own data to train the mobilenet ssd model ,everything is fine,the model size is 22.2MB, but something is wrong when i run the demo ,the GPU memory is 700M, and the spend time is 270ms /image,so could you tell me the reason, looking forward your reply!!

ZhangCats commented 4 years ago

when you run demo.py,you should put : caffe.set_mode_gpu() after import caffe