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/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)