pcuenq/lpips-jax

Weights for JAX/Flax version of VGG

• —VGG16 weights, taken from the `flaxmodels` repo.
• —Additional weights to use VGG16 as a feature extractor for LPIPS. They were downloaded in PyTorch format from the URL referenced in the Taming Transformers repo, and converted to hdf5 format.

License

Apache 2, for this compilation.

Please, refer to the original licenses of the source repos.

• —Taming Transformers License. Weights for additional layers.
• —Perceptual Similarity License. Weights for additional layers.
• —Flaxmodels / VGG License, for the VGG model and (I presume) VGG weights.