search

keras-team / keras-contrib

Keras community contributions
MIT License
1.58k stars 650 forks source link

Incorporate more Models and ImageNet pretrained weights #123

Open ahundt opened 7 years ago

ahundt commented 7 years ago

https://github.com/flyyufelix/cnn_finetune has densenet and resnet weights

ahundt commented 7 years ago

@titu1994 you may be interested in the linked repo

titu1994 commented 7 years ago

Thanks for the heads up! @ahundt.

ResNet weights were already available, but DenseNet weights may be worth porting over. Is the model compatible ? Sadly I don't have time right now to look into porting this weights.

I'd like to take a look at this in a month or so, when I have some time.

titu1994 commented 7 years ago

The code is very similar thankfully. The Scale layer is the only major difference. Perhaps someone can port over this code and the weight loading procedure to this repository ?

ahundt commented 7 years ago

I've been thinking of doing that but my plate is full trying to get the tfrecord changes in upstream, which is why I just created the issue for now.

Also the repo has versions of resnet not currently supported in keras.