compatible with DLC3.0

Mojackhak · Oct 12, 2024 · 1483500 · 1483500
1 parent cb9df8c
commit 1483500
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/anipose/pose_videos.py b/anipose/pose_videos.py
@@ -14,7 +14,11 @@ def rename_dlc_files(folder, base):
     files = glob(os.path.join(folder, base+'*'))
     for fname in files:
         basename = os.path.basename(fname)
+        # compatibility with DLC 3.0
         _, ext = os.path.splitext(basename)
+        if ext != '.h5' and ext != '.csv':
+            _, ext = basename.rsplit('_', 1)
+            ext = '_' + ext
         os.rename(os.path.join(folder, basename),
                   os.path.join(folder, base + ext))