mask tool: log and exit if batch not using folder

deepfakes · Jun 30, 2023 · 8fb96b0 · 8fb96b0
1 parent 4e7e5fe
commit 8fb96b0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tools/mask/mask.py b/tools/mask/mask.py
@@ -59,6 +59,10 @@ def _get_input_locations(self) -> list[str]:
         if not self._args.batch_mode:
             return [self._args.input]
 
+        if not os.path.isdir(self._args.input):
+            logger.error("Batch mode is selected but input '%s' is not a folder", self._args.input)
+            sys.exit(1)
+
         retval = [os.path.join(self._args.input, fname)
                   for fname in os.listdir(self._args.input)
                   if os.path.isdir(os.path.join(self._args.input, fname))