bugfix: Patch writer - handle filenames which are only numeric

deepfakes · Jan 10, 2024 · b25f484 · b25f484
1 parent 2a38451
commit b25f484
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/plugins/convert/writer/patch.py b/plugins/convert/writer/patch.py
@@ -95,6 +95,8 @@ def _get_new_filename(self, filename: str, face_index: int) -> str:
         """
         face_idx = str(face_index).rjust(2, "0")
         fname, ext = os.path.splitext(filename)
+        fname = os.path.basename(fname)
+
         split_fname = fname.rsplit("_", 1)
         if split_fname[-1].isdigit():
             i_frame_no = (int(split_fname[-1]) +