search

keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras
Other
1.01k stars 330 forks source link

Migrate ConvMixer to the new `Backbone` API #1638

Closed ianstenbit closed 1 year ago

ianstenbit commented 1 year ago

To align KerasCV with the KerasNLP API, we're migrating our backbone models to subclass keras_cv.models.Backbone. This allows us to have a more structured, modular API for high-level task models, including the new ImageClassifier model (see https://github.com/keras-team/keras-cv/pull/1458).

A great step-by-step guide to this migration can be found in the description of https://github.com/keras-team/keras-cv/pull/1508, which performed this migration for ResNetV1.

If you'd like to take this issue up, please leave a comment and the KerasCV team can assign the issue to you. When opening a PR, please tag this issue as fixed by your PR, and tag our tracking bug https://github.com/keras-team/keras-cv/issues/1627 for bookkeeping.

ID6109 commented 1 year ago

I'd like to take this up. I've worked on ConvMixer in #1476

ianstenbit commented 1 year ago

We're most likely not going to move forward with ConvMixer as a backbone due to low interest