lpips-j – Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.
Project description
LPIPS-J
This is a minimal JAX/Flax port of lpips, as implemented in:
Only the essential features have been implemented. Our motivation is to support VQGAN training for DALL•E Mini.
It currently supports the vgg16 backend, leveraging the implementation in flaxmodels.
Pre-trained weights for the network and the linear layers are downloaded from the 🤗 Hugging Face hub.
Installation
- Install JAX for CUDA or TPU following the instructions at https://github.com/google/jax#installation.
- Install this package from the repository:
pip install --upgrade git+https://github.com/pcuenca/lpips-j.git
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
lpips-j-0.0.6.tar.gz
(7.2 kB
view details)
Built Distribution
File details
Details for the file lpips-j-0.0.6.tar.gz.
File metadata
- Download URL: lpips-j-0.0.6.tar.gz
- Upload date:
- Size: 7.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 413ef9592d825998b53c86d8d249e0b0177c7a3ffeeb7fe2fb4ac67b57700353 |
|
| MD5 | db06b62f407c6412d1feef33108a6a58 |
|
| BLAKE2b-256 | f1dc78097a26296b01a41f88d494abce2ec262edf6b45534620250e24d1a00f9 |
File details
Details for the file lpips_j-0.0.6-py3-none-any.whl.
File metadata
- Download URL: lpips_j-0.0.6-py3-none-any.whl
- Upload date:
- Size: 7.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | a408842d9719521e78349202732d71859f15b68172f20aafb08354cd028c5c02 |
|
| MD5 | 9d429d125fc7d9ac2af43670f892e112 |
|
| BLAKE2b-256 | 6d402f48e0cbaf0cedc9b788503ab89a5dcbb0537caed086d421c678da142da4 |
