search

daerduoCarey / SpatialTransformerLayer

Other
181 stars 97 forks source link

.caffemodel for initialisation #9

Closed Tushar-N closed 7 years ago

Tushar-N commented 7 years ago

Hi,

When you trained your models, did you initialise the inception networks with imagenet-pretrained weights or did you train from scratch? I ask because in your solver.prototxt you load weights from "models/googlenet-bn-cub-st/init_googlenet_bn_cub_two_st.caffemodel". Could you share the .caffemodel that you used please?

Thank you.

daerduoCarey commented 7 years ago

If I remember correctly, I used the pretrained inception network from caffe model zoo. Since in the cub experiment, the network is quite large so it's pretty hard to train from the scratch. Hope this answer your question.

For the example script, "init_googlenet_bn_cub_two_st.caffemodel" is the intermediate caffemodel that I generated in my experiment. Don't use that. My script just shows you how to use the layer and please replace it with your own pretrained model or officially released pretrained models from caffe model zoo or other resources.

Thank you.

Tushar-N commented 7 years ago

Got it, thank you!

On Mar 8, 2017 11:36 AM, "Kaichun Mo" notifications@github.com wrote:

If I remember correctly, I used the pretrained inception network from caffe model zoo. Since in the cub experiment, the network is quite large so it's pretty hard to train from the scratch. Hope this answer your question.

For the example script, "init_googlenet_bn_cub_two_st.caffemodel" is the intermediate caffemodel that I generated in my experiment. Don't use that. My script just shows you how to use the layer and please replace it with your own pretrained model or officially released pretrained models from caffe model zoo or other resources.

Thank you.

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/daerduoCarey/SpatialTransformerLayer/issues/9#issuecomment-285111040, or mute the thread https://github.com/notifications/unsubscribe-auth/AEw4GTePiH4fKNCYlb4yn0hNeg0GbvBCks5rjuckgaJpZM4L-ASi .

sunformoon commented 7 years ago

@daerduoCarey Hi Kaichun, thank you so much for your efforts! I am wondering if you could share the pretrained models? like, init_CUB_googLeNet_ST_INC1_INC2.caffemodel, init_googlenet_bn_cub_two_st.caffemodel, etc. Much appreciated!

Best, Zhuotun