github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

basic hack for osx

Browse source
This commit is contained in:
jkennedyvz 2022-05-22 18:02:03 -07:00
parent 0d19d8ec8e
commit 8aadbfef4a
3 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/imagelib/warp.py
View file

@ -143,7 +143,7 @@ def gen_warp_params (w, flip=False, rotation_range=[-10,10], scale_range=[-0.5,
################
#random transform
random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale)
random_transform_mat = cv2.getRotationMatrix2D((int(w // 2), int(w // 2)), rotation, scale)
random_transform_mat[:, 2] += (tx*w, ty*w)
params = dict()

4
core/leras/device.py
View file

@ -46,7 +46,7 @@ class Devices(object):
idx_mem = 0
for device in self.devices:
mem = device.total_mem
if mem > idx_mem:
if mem >= idx_mem:
result = device
idx_mem = mem
return result
@ -56,7 +56,7 @@ class Devices(object):
idx_mem = sys.maxsize
for device in self.devices:
mem = device.total_mem
if mem < idx_mem:
if mem <= idx_mem:
result = device
idx_mem = mem
return result

1
core/leras/nn.py
View file

@ -112,6 +112,7 @@ class nn():
config.gpu_options.force_gpu_compatible = True
config.gpu_options.allow_growth = True
config.allow_soft_placement = True
nn.tf_sess_config = config
if nn.tf_sess is None: