flax-vision-models 0.0.3

A repository of Deep Learning models in Flax

Flax Vision Models

A repository of Deep Learning models in Flax pre-trained on Imagenet. Most of the models instead of being trained from scratch have been ported from exiting repositories such as tf.keras.applications and PyTorch Image Models.

Usage

Installation

Sample Inference Example

Pre-trained Models

VGG Models

Model	Original Source	Paper
flax_models.vgg.build_vgg16	tf.keras.applications.vgg16.VGG16	Very Deep Convolutional Networks for Large-Scale Image Recognition
flax_models.vgg.build_vgg19	tf.keras.applications.vgg19.VGG19	Very Deep Convolutional Networks for Large-Scale Image Recognition