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

fix cvae

Browse Source
This commit is contained in:
luni
2023-01-26 14:45:35 +00:00
parent ea683ac030
commit 5d91b56a63
5 changed files with 34 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/base.py
+22 -6
View File
@@ -118,13 +118,29 @@ class VariationalEncoderNew(tf.keras.Model):
self.encoder = resnet_18(features)
self.mean_branch = make_basic_block_layer(filter_num=features,
blocks=2,
stride=1)
self.mean_branch = tf.keras.Sequential([
tf.keras.layers.Conv2D(filters=features,
kernel_size=(3, 3),
strides=1,
padding="same"),
tf.keras.layers.BatchNormalization(),
tf.keras.layers.Conv2D(filters=features,
kernel_size=(3, 3),
strides=1,
padding="same")
])
self.var_branch = make_basic_block_layer(filter_num=features,
blocks=2,
stride=1)
self.var_branch = tf.keras.Sequential([
tf.keras.layers.Conv2D(filters=features,
kernel_size=(3, 3),
strides=1,
padding="same"),
tf.keras.layers.BatchNormalization(),
tf.keras.layers.Conv2D(filters=features,
kernel_size=(3, 3),
strides=1,
padding="same")
])
self.flatten_mean = tf.keras.layers.Flatten()
self.flatten_var = tf.keras.layers.Flatten()
models/cvae_model.py
+1 -1
View File
@@ -182,7 +182,7 @@ class CVAENew(tf.keras.Model):
self.encoder = VariationalEncoderNew(config['cvae']['encoder_dims'][-1])
self.decoder = DecoderNew(config['cvae']['decoder_dims'][-1], is_cvae=True)
self.reshape = tf.keras.layers.Reshape((16, 8, config['cvae']['decoder_dims'][-1]))
self.reshape = tf.keras.layers.Reshape((8, 4, config['cvae']['decoder_dims'][-1]))
if config['avg_garment_path'] != '':
self.avg_garment = np.load(config['avg_garment_path'])