This commit is contained in:
Colombo 2020-01-22 18:49:32 +04:00
parent 2d61b715de
commit 0ce96875a1
2 changed files with 3 additions and 2 deletions

View file

@ -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)