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

commit all

Browse Source
This commit is contained in:
luni
2023-01-25 12:25:59 +00:00
parent 1bf224841f
commit 7470ca9fb6
139 changed files with 82069 additions and 811 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/pyramid_model.py
+4 -4
View File
@@ -49,7 +49,7 @@ class Pyramid(tf.keras.Model):
load_model(self.upscale_enhancer, config['pyramid']['final_upscale_enhancer'])
self.upscale_enhancer.trainable = False
def enhance(self, low, mask, training=None):
def enhance(self, low, mask, training):
x = self.upscale_enhancer([low, mask], training=training)
return x
@@ -63,7 +63,7 @@ class Pyramid(tf.keras.Model):
return z_cond
def call(self, inputs, training=None):
def call(self, inputs, training):
levels_input, conds = inputs
uv_static, uv_body, uv_curv = conds[0], conds[1], conds[2]
z_cond = self.encode_conds(uv_static, uv_body, uv_curv)
@@ -136,7 +136,7 @@ class Pyramid(tf.keras.Model):
# load_model(self.upscale_enhancer, config['pyramid']['final_upscale_enhancer'])
# self.upscale_enhancer.trainable = False
# def enhance(self, low, mask, training=None):
# def enhance(self, low, mask, training):
# x = self.upscale_enhancer([low, mask], training=training)
# return x
@@ -150,7 +150,7 @@ class Pyramid(tf.keras.Model):
# return z_cond
# def call(self, inputs, training=None):
# def call(self, inputs, training):
# levels_input, conds = inputs
# uv_static, uv_body, uv_curv = conds[0], conds[1], conds[2]
# z_cond = self.encode_conds(uv_static, uv_body, uv_curv)