Pedro Cuenca
commited on
Commit
·
1a6ac74
1
Parent(s):
c98375d
Add weights, update README.
Browse files- README.md +15 -0
- lpips_lin.h5 +3 -0
- vgg16_weights.h5 +3 -0
README.md
CHANGED
|
@@ -1,3 +1,18 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
---
|
| 4 |
+
|
| 5 |
+
## Weights for JAX/Flax version of VGG
|
| 6 |
+
|
| 7 |
+
- VGG16 weights, taken from [the `flaxmodels` repo](https://github.com/matthias-wright/flaxmodels/blob/main/flaxmodels/vgg/vgg.py).
|
| 8 |
+
- 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.
|
| 9 |
+
|
| 10 |
+
## License
|
| 11 |
+
|
| 12 |
+
Apache 2, for this compilation.
|
| 13 |
+
|
| 14 |
+
Please, refer to the original licenses of the source repos.
|
| 15 |
+
- [Taming Transformers License](https://github.com/CompVis/taming-transformers/blob/master/License.txt). Weights for additional layers.
|
| 16 |
+
- [Perceptual Similarity License](https://github.com/richzhang/PerceptualSimilarity/blob/master/LICENSE). Weights for additional layers.
|
| 17 |
+
- [Flaxmodels / VGG License](https://github.com/matthias-wright/flaxmodels/tree/main/flaxmodels/vgg#license), for the VGG model and (I presume) VGG weights.
|
| 18 |
+
|
lpips_lin.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7d45016fac238ac6d2132412210e6ecfbbe49803226ac9d5f1d98ed06dab7d31
|
| 3 |
+
size 10240
|
vgg16_weights.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3fef947be1d1fa614be89033a66945d0645fd0eb6dd88d9ed213b322efd8f75c
|
| 3 |
+
size 553459416
|