search

titu1994 / keras-squeeze-excite-network

Implementation of Squeeze and Excitation Networks in Keras
MIT License
400 stars 118 forks source link

Any plans to support pretrained models? #3

Closed antorsae closed 6 years ago

antorsae commented 6 years ago

The function calls includes a weights parameter to optionally support imagenet pre-trained weights, however there's no real loading of such weights.

titu1994 commented 6 years ago

Tuesday original weights are in Torch / Caffe format. I use Windows so I cannot use those two libraries, and therefore cannot port weights.

I will ofcourse welcome PRs for weights.