Open ahundt opened 7 years ago
@titu1994 you may be interested in the linked repo
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.
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 ?
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.
https://github.com/flyyufelix/cnn_finetune has densenet and resnet weights