ssds.pytorch icon indicating copy to clipboard operation
ssds.pytorch copied to clipboard

I train a new model,but test speed is too slow?

Open funnyuday opened this issue 6 years ago • 5 comments

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!

funnyuday avatar May 05 '18 09:05 funnyuday

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,

foreverYoungGitHub avatar May 09 '18 16:05 foreverYoungGitHub

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 avatar May 10 '18 09:05 funnyuday

@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!

nekulkarni avatar Aug 29 '18 01:08 nekulkarni

@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 avatar Apr 10 '19 06:04 blueardour

@blueardour I used the early training model to test.It need too many boxes to NMS.

funnyuday avatar Apr 10 '19 09:04 funnyuday