I’m sharing a personal project of mine, which was to rewrite ResNet-RS models from TPUEstimator to Tensorflow/Keras.
Features:
Automatic weights download.
Transfer learning possible. pip install directly from GitHub. keras.applications like usage.
Use like any other Tensorflow/Keras model!
@sebastian-sz cc. @Sayak_Paul
Amazing job. There’s an option to contribute to the core API of these model families (as mentioned by brother sayak). If you consider that would be great, like you did for EfficientNet-V2
I quickly took a look at your implementation and it seems very structured. I think barring a few formatting and Keras-specific nits, you’d be well on your way.
@Bhack
Do you mean that the PR, regarding this model family, should be opened in keras-cv instead of keras? Or that ResNetRS models should be retrained from scratch with components created in keras-cv?
The repository I created is simply a port of model architecture + weights. I did not include training code (but also none of the models under keras.applications do).
As for training components like dropblock / drop connect: they are present in the original repository but upon closer inspection their parameters are always either 0 or null, so I decided not to include them.