Flax-vision-models

A repository of Deep Learning models in [Flax](https://github.com/google/flax) pre-trained on [Imagenet](https://image-net.org/). Most of the models instead of being trained from scratch have been ported from exiting repositories such as [`tf.keras.applications`](https://www.tensorflow.org/api_docs/python/tf/keras/applications) and [PyTorch Image Models](https://github.com/rwightman/pytorch-image-models).

Usage

Installation


Sample Inference Example


Pre-trained Models

VGG Models

|Model|Original Source|Paper|
|---|---|---|

A repository of Deep Learning models in [Flax](https://github.com/google/flax) pre-trained on [Imagenet](https://image-net.org/). Most of the models instead of being trained from scratch have been ported from exiting repositories such as [`tf.keras.applications`](https://www.tensorflow.org/api_docs/python/tf/keras/applications) and [PyTorch Image Models](https://github.com/rwightman/pytorch-image-models).

Pre-trained Models

VGG Models

|Model|Original Source|Paper|
|---|---|---|

VGG16 and VGG19 models converted from [`tf.keras.applications`](https://www.tensorflow.org/api_docs/python/tf/keras/applications).

|Model|Original Source|Paper|
|---|---|---|