luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

run-35

Browse Source
This commit is contained in:
luni
2022-06-27 12:27:55 +00:00
parent 5e00b395ad
commit cd498e79f3
11 changed files with 301 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files
stylegan2/layers/embedding.py
+1
View File
@@ -97,6 +97,7 @@ class GarmentEmbedding(tf.keras.layers.Layer):
def call(self, inputs, training=None, mask=None):
x = inputs
x = tf.image.per_image_standardization(x)
x = tf.transpose(x, [0,2,3,1])
x = self.encoder(x, training=training)
stylegan2/layers/mapping_block.py
+2 -2
View File
@@ -31,8 +31,8 @@ class Mapping(tf.keras.layers.Layer):
x = latents
# embed label if any
pose_shape_latent = self.pose_shape_embedding([conditions[0], conditions[1]])
garment_latent = self.garment_embedding(conditions[2])
pose_shape_latent = self.pose_shape_embedding([conditions['pose_shape'][:, :72], conditions['pose_shape'][:, 72:]])
garment_latent = self.garment_embedding(conditions['uv_static'])
x = tf.concat([x, pose_shape_latent, garment_latent], axis=1)
# x = tf.concat([x, garment_latent], axis=1)