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

fix multigpu extractor

Browse source
This commit is contained in:
Colombo 2019-10-10 13:01:57 +04:00
parent 9b08199e87
commit d2d0dcb5e7
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
nnlib/device.py
View file

@ -102,7 +102,7 @@ class device:
elif device.backend == "tensorflow": elif device.backend == "tensorflow":
for dev in cuda_devices: for dev in cuda_devices:
if dev['total_mem'] >= totalmemsize_gb*1024*1024*1024: if dev['total_mem'] >= totalmemsize_gb*1024*1024*1024:
result.append (i) result.append ( dev['index'] )
return result return result

2
samplelib/SampleProcessor.py
View file

@ -260,7 +260,7 @@ class SampleProcessor(object):
elif mode_type == SPTF.MODE_G: elif mode_type == SPTF.MODE_G:
img = np.concatenate ( (np.expand_dims(cv2.cvtColor(img_bgr, cv2.COLOR_BGR2GRAY),-1),img_mask) , -1 ) img = np.concatenate ( (np.expand_dims(cv2.cvtColor(img_bgr, cv2.COLOR_BGR2GRAY),-1),img_mask) , -1 )
elif mode_type == SPTF.MODE_GGG: elif mode_type == SPTF.MODE_GGG:
img = np.concatenate ( ( np.repeat ( np.expand_dims(cv2.cvtColor(img_bgr, cv2.COLOR_BGR2GRAY),-1), (3,), -1), img_mask), -1) img = np.repeat ( np.expand_dims(cv2.cvtColor(img_bgr, cv2.COLOR_BGR2GRAY),-1), (3,), -1)
elif mode_type == SPTF.MODE_M and is_face_sample: elif mode_type == SPTF.MODE_M and is_face_sample:
img = img_mask img = img_mask