github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-21 05:53:24 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

upgrade to tf 2.4.0rc1

Browse source
This commit is contained in:
Colombo 2020-11-13 17:00:07 +04:00
commit 0eb7e06ac1
6 changed files with 23 additions and 59 deletions
Download patch file Download diff file Expand all files Collapse all files

10
core/leras/ops/__init__.py
View file

@ -333,7 +333,17 @@ def depth_to_space(x, size):
x = tf.reshape(x, (-1, oh, ow, oc, ))
return x
else:
b,c,h,w = x.shape.as_list()
oh, ow = h * size, w * size
oc = c // (size * size)
x = tf.reshape(x, (-1, size, size, oc, h, w, ) )
x = tf.transpose(x, (0, 3, 4, 1, 5, 2))
x = tf.reshape(x, (-1, oc, oh, ow))
return x
return tf.depth_to_space(x, size, data_format=nn.data_format)
nn.depth_to_space = depth_to_space
def rgb_to_lab(srgb):