mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 04:52:13 -07:00
1
This commit is contained in:
parent
2d61b715de
commit
0ce96875a1
2 changed files with 3 additions and 2 deletions
|
@ -55,8 +55,6 @@ class ExtractSubprocessor(Subprocessor):
|
|||
if stdin_fd is not None and DEBUG:
|
||||
sys.stdin = os.fdopen(stdin_fd)
|
||||
|
||||
self.log_info (f"Running on {client_dict['device_name'] }")
|
||||
|
||||
if self.cpu_only:
|
||||
device_config = nn.DeviceConfig.CPU()
|
||||
place_model_on_cpu = True
|
||||
|
@ -67,6 +65,8 @@ class ExtractSubprocessor(Subprocessor):
|
|||
if self.type == 'all' or 'rects' in self.type or 'landmarks' in self.type:
|
||||
nn.initialize (device_config)
|
||||
|
||||
self.log_info (f"Running on {client_dict['device_name'] }")
|
||||
|
||||
if self.type == 'all' or self.type == 'rects-s3fd' or 'landmarks' in self.type:
|
||||
self.rects_extractor = facelib.S3FDExtractor(place_model_on_cpu=place_model_on_cpu)
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue