Pedro Cuenca commited on
Commit
1a6ac74
·
1 Parent(s): c98375d

Add weights, update README.

Browse files
Files changed (3) hide show
  1. README.md +15 -0
  2. lpips_lin.h5 +3 -0
  3. 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