realtime_object_detection icon indicating copy to clipboard operation
realtime_object_detection copied to clipboard

Train SSD 600

Open paolomanchisi opened this issue 6 years ago • 1 comments

HI, I want to train my custom dataset with ssd 600. Is it enough to set

 image_resizer {
      fixed_shape_resizer {
        height: 300
        width: 300
      }
    }

to 600 / 600 instead of 300 / 300 ? Thank you very much.

paolomanchisi avatar Apr 11 '18 14:04 paolomanchisi

yes that is enough, but then you won't be able to use the pre-trained checkpoints. As the weights only fit a model that takes the same input dimensions!

EDIT: Infact you are able to use the pre-trained weights, but you have to set from_detection_checkpoint to false

gustavz avatar Apr 25 '18 06:04 gustavz