keras-team / keras-cv

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

Add support of PyTorch and Jax to EfficentNetV1, EfficientNetLite, and RegNet Model #1941

Open cosmo3769 opened 1 year ago

cosmo3769 commented 1 year ago

Since the three models is almost completely converted to backbone.

I am willing to add support of Pytorch and JAX to these models as well. Should I add it in these created PR only or should I create another PR for this?

@ianstenbit @jbischof @LukeWood

ianstenbit commented 1 year ago

Let's do it in the open PR that you have -- let me know if you have any trouble converting the backbones.

Here's an example PR you can use to see how to add Keras Core support: https://github.com/ianstenbit/keras-cv/pull/4/files