Metadata-Version: 2.1
Name: flax-vision-models
Version: 0.0.2
Summary: A repository of Deep Learning models in Flax
Home-page: https://github.com/soumik12345/flax-vision-models
Author: Soumik Rakshit
Author-email: 19soumik.rakshit96@gmail.com
License: UNKNOWN
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Requires-Python: >=3.7.12,<3.11
Description-Content-Type: text/markdown

# 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

![](assets/1.svg)

### Sample Inference Example

![](assets/2.svg)

# Pre-trained Models

## VGG Models

|Model|Original Source|Paper|
|---|---|---|
|`flax_models.vgg.build_vgg16`|[`tf.keras.applications.vgg16.VGG16`](https://github.com/keras-team/keras/blob/v2.8.0/keras/applications/vgg16.py#L43-L227)|[Very Deep Convolutional Networks for Large-Scale Image Recognition](https://arxiv.org/abs/1409.1556)|
|`flax_models.vgg.build_vgg19`|[`tf.keras.applications.vgg19.VGG19`](https://github.com/keras-team/keras/blob/v2.8.0/keras/applications/vgg19.py#L43-L231)|[Very Deep Convolutional Networks for Large-Scale Image Recognition](https://arxiv.org/abs/1409.1556)|

