lpips-jax / README.md
Pedro Cuenca
Add weights, update README.
1a6ac74
---
license: apache-2.0
---
## Weights for JAX/Flax version of VGG
- VGG16 weights, taken from [the `flaxmodels` repo](https://github.com/matthias-wright/flaxmodels/blob/main/flaxmodels/vgg/vgg.py).
- 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](https://github.com/CompVis/taming-transformers/blob/master/taming/modules/losses/lpips.py), and converted to hdf5 format.
## License
Apache 2, for this compilation.
Please, refer to the original licenses of the source repos.
- [Taming Transformers License](https://github.com/CompVis/taming-transformers/blob/master/License.txt). Weights for additional layers.
- [Perceptual Similarity License](https://github.com/richzhang/PerceptualSimilarity/blob/master/LICENSE). Weights for additional layers.
- [Flaxmodels / VGG License](https://github.com/matthias-wright/flaxmodels/tree/main/flaxmodels/vgg#license), for the VGG model and (I presume) VGG weights.