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

activate all levels

Browse Source
This commit is contained in:
luni
2023-03-15 22:21:36 +01:00
parent 7ee10aa7a2
commit 319dc9372b
2 changed files with 99 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/pyramid_model_optimized.py
+1 -1
View File
@@ -163,8 +163,8 @@ class PyramidNew(tf.keras.Model):
pred = pred + levels_input[0][2] # uv_body_posed_32
if 1 != len(self.levels):
pred = self.levels[0].enhance(pred, levels_input[0][-2], training=training) # levels_input[i][-2]=uv_mask_res
pred = tf.image.resize(pred, (self.levels[1].size, self.levels[1].size//2), tf.image.ResizeMethod.BILINEAR)
out_levels.append([pred, mean, logvar])
pred = tf.image.resize(pred, (self.levels[1].size, self.levels[1].size//2), tf.image.ResizeMethod.BILINEAR)
pred = pred * levels_input[1][-2][:,:,:,tf.newaxis]
else:
# pred = self.enhance(pred, levels_input[0][-2], training=training)