diff --git a/mainscripts/Extractor.py b/mainscripts/Extractor.py index d2aa4e5..30e629b 100644 --- a/mainscripts/Extractor.py +++ b/mainscripts/Extractor.py @@ -193,6 +193,9 @@ class ExtractSubprocessor(Subprocessor): io.progress_bar_close() def get_devices_for_type (self, type, multi_gpu, cpu_only): + if 'cpu' in nnlib.device.backend: + cpu_only = True + if not cpu_only and (type == 'rects' or type == 'landmarks'): if type == 'rects' and self.detector == 'mt' and nnlib.device.backend == "plaidML": cpu_only = True