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/upscaler.py
+6 -6
View File
@@ -12,7 +12,7 @@ class Bottleneck(tf.keras.Model):
layer_scale_init_value=1e-6, prefix=f'bottleneck{j}') for j in range(5)]
)
def call(self, inputs, training=None):
def call(self, inputs, training):
x = inputs
x = self.bottleneck(x, training=training)
@@ -35,7 +35,7 @@ class Upscaler(tf.keras.Model):
self.decoder = ConvNeXtReverseWithSkip(depths=depths, dims=reverse_dims, kernel_size=3)
def call(self, inputs, training=None):
def call(self, inputs, training):
img, mask = inputs
img = tf.transpose(img, [0,2,3,1])
@@ -74,7 +74,7 @@ class UpscalerBasic(tf.keras.Model):
self.reshape = tf.keras.layers.Reshape((low_res, (low_res//2), 3))
def call(self, inputs, training=None):
def call(self, inputs, training):
img, mask = inputs
x = self.flatten(img)
@@ -110,7 +110,7 @@ class UpscalerBasicNew(tf.keras.Model):
self.reshape = tf.keras.layers.Reshape((low_res, (low_res//2), 3), name='upscale_enhancer_reshape')
def call(self, inputs, training=None):
def call(self, inputs, training):
img, mask = inputs
x = self.flatten(img)
@@ -143,7 +143,7 @@ class UpscalerBasicConv(tf.keras.Model):
])
def call(self, inputs, training=None):
def call(self, inputs, training):
img, mask = inputs
x = self.seq(img, training=training)
@@ -160,7 +160,7 @@ class UpscalerBasicConv(tf.keras.Model):
# self.decoder = ConvNeXtReverse(depths=list(reversed([9, 9, 9])), dims=list(reversed([bottleneck_size, 128, 32, 16])))
# def call(self, inputs, training=None):
# def call(self, inputs, training):
# x, mask = inputs
# x = tf.transpose(x, [0,2,3,1])