keras-squeeze-excite-network icon indicating copy to clipboard operation
keras-squeeze-excite-network copied to clipboard

Pre-trained model

Open cipher009 opened this issue 6 years ago • 5 comments

Does anyone have pre-trained weights available?

cipher009 avatar Apr 05 '18 09:04 cipher009

@cipher009 how to pre-trained the model? I check the keras/example/application and it show me how to pre-trained model. i check the se_densenet.py.

model = sedensenet121 (include_top = false,weights = 'imagenet')

but it show me an error.

alyato avatar Jun 26 '18 01:06 alyato

If you check the code it actually never tries to load imagenet weights for most of the densenets.

Does anyone know if you can train these networks directly on ~50k images or do i need to get the imagenet dataset and 'burn it in' with this? Also, does anyone know how long it would take on !one! gtx1080ti?

shsshs avatar Nov 30 '18 10:11 shsshs

Does anyone find the pre-trained weights?

Fahim-F avatar Jan 30 '19 19:01 Fahim-F

The Se_ResNet script never actually loads weights.

mikelam14 avatar Mar 06 '19 20:03 mikelam14

Found weights for ResNet. https://github.com/qubvel/classification_models/blob/master/classification_models/weights.py

You will have to downgrade Keras version to 2.1.2 to run it.

mikelam14 avatar Mar 06 '19 21:03 mikelam14