extrapolation train

2023-05-25 18:09:55 +02:00
parent 843c9ac4b0
commit af9811e731
7 changed files with 596 additions and 13 deletions
@@ -45,7 +45,10 @@ def read_frames(folder, shuffle=True, is_cape=False, testing=False, frames_file=
        with open(os.path.join(folder, frames_file), 'r') as f:
            lines = f.readlines()
            lines = [line.strip() for line in lines]
-        frames = [Frame(line.split('_')[0], line.split('_')[1]) for line in lines]
+        if is_cape:
+            frames = [Frame(line.split('.')[0], line.split('.')[1]) for line in lines]
+        else:
+            frames = [Frame(line.split('_')[0], line.split('_')[1]) for line in lines]

    if shuffle:
        random.shuffle(frames)