ssds.pytorch
ssds.pytorch copied to clipboard
I train a new model,but test speed is too slow?
I down load the mobilenet_v1_ssd_voc_72.7, it's test speed faster(0.03s) than my new model(0.3s).
I can't find the reason,the configuration file is same as mobilenet_v1_ssd_voc_72.
I also train new mobilenetv2-ssd in same file, the speed is still very slow. Please give me some advice. Thank you very much!
I just update the time_benchmark part and it will record the timing into a csv file. Would you mind to run the code again and share the csv file?
Best,
Thank you for your reply,I have solved this problem. My model spend more time on running NMS. I fine-tune the mobilenet_v1_ssd_voc_72.7.The speed is very fast.
@funnyuday How did you finetune? I want to finetune the same model with my custom dataset as well, but I can't find out how to do that. Thanks in advance!
@funnyuday I found a similar problem. The NMS is very very slow on my machine. Could you share what you have revised to improve the inference detection speed?
@blueardour I used the early training model to test.It need too many boxes to NMS.