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

enhancing landmarks extractor by using s3fd second pass inside second pass,

Browse source 
it will be x2 slower, but time will be saved due to more images will be marked properly
works on 2GB+
This commit is contained in:
iperov 2019-03-18 10:25:24 +04:00
parent 1d56585f33
commit b8efb4cbba
3 changed files with 51 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

7
nnlib/device.py
View file

@ -64,6 +64,7 @@ class device:
self.cpu_only = (len(self.gpu_idxs) == 0)
if not self.cpu_only:
self.gpu_names = []
self.gpu_compute_caps = []
@ -73,7 +74,11 @@ class device:
self.gpu_compute_caps += [ device.getDeviceComputeCapability(gpu_idx) ]
self.gpu_vram_gb += [ device.getDeviceVRAMTotalGb(gpu_idx) ]
self.cpu_only = (len(self.gpu_idxs) == 0)
else:
self.gpu_names = ['CPU']
self.gpu_compute_caps = [99]
self.gpu_vram_gb = [0]
if self.cpu_only:
self.backend = "tensorflow-cpu"