caffe icon indicating copy to clipboard operation
caffe copied to clipboard

How to modify the model in order to train a ssd with 150*150 inputs?

Open MinLiAmoy opened this issue 7 years ago • 6 comments

Hi, recently I am considering to train a model and the size of the input image is 150*150. I expect the speed could be faster than ssd300, due to the computation and smaller feature map to generate bbox. How can I modify the code or prototxt? Thanks!

MinLiAmoy avatar Jan 18 '18 16:01 MinLiAmoy

check out train.prototxt file , you need to change dimensions there, likewise for test and deploy prototxt

linux-devil avatar Jan 19 '18 12:01 linux-devil

@linux-devil thanks for your reply =)

MinLiAmoy avatar Jan 24 '18 05:01 MinLiAmoy

You may close this issue

linux-devil avatar Jan 24 '18 07:01 linux-devil

I changed train.prototxt test.prototxt deploy.prototxt resize_param or dimensions from 300 to 150 and then train again but the result of detect is very bad. Do you still need to modify other parameters?

rmx564354093 avatar May 02 '18 03:05 rmx564354093

I changed train.prototxt test.prototxt deploy.prototxt resize_param or dimensions from 300 to 150 and then train again but the result of detect is very bad. Do you still need to modify other parameters?

change min_size and max_size in priorbox layer

yokings avatar Nov 06 '18 08:11 yokings

I am facing the same issue. Updated the min_size and max_size in priorbox layer but the performance is still bad. On the same dataset, the SSD model with image size 300x300 performs very well. Any reasons why on 150x150, it is not training?

soumilchugh avatar Nov 09 '20 15:11 soumilchugh