Skip to main content

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

Release files for flax-vision-models 0.0.3

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for flax-vision-models 0.0.3
File Size Uploaded
flax-vision-models-0.0.3.tar.gz 4.9 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for flax-vision-models 0.0.3
File Interpreter ABI Platform
flax_vision_models-0.0.3-py3-none-any.whl Python 3 none any Details

Total release size: 11.1 kB

Release files / flax-vision-models-0.0.3.tar.gz

Download URL flax-vision-models-0.0.3.tar.gz
Size 4.9 kB
Tags Source
SHA-256 checksum
How to use checksums
3bb53efa2f34584cbadb32e04e5155e62c78e544fa356f4ec3354c2013ec3592
BLAKE2b-256 checksum
How to use checksums
72974801d5db0e28dba7ea4e698796da7c78b913d19bd99a761a1d4ee98d7aa2
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.0 CPython/3.10.4

Release files / flax_vision_models-0.0.3-py3-none-any.whl

Download URL flax_vision_models-0.0.3-py3-none-any.whl
Size 6.2 kB
Tags Python 3
SHA-256 checksum
How to use checksums
8619c6476e50362f9b0334f363d0d11c085ee9edf508ff7fa4673a05e3caed15
BLAKE2b-256 checksum
How to use checksums
c35836daed067a458ac5e17d1722364ec6113a38d8a8ecc6827e13f477f23a2c
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.0 CPython/3.10.4

Release history Release notifications | RSS feed

This release

0.0.3 This release

2 release files

0.0.2

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page