Spaces:
Runtime error
Runtime error
Update utils.py
Browse files
utils.py
CHANGED
@@ -41,8 +41,8 @@ def load_data(path: str):
|
|
41 |
file_name = path.split('/')[-1].split('.')[0]
|
42 |
# File name splitting for windows
|
43 |
file_name = path.split('\\')[-1].split('.')[0]
|
44 |
-
video_path = os.path.join('
|
45 |
-
alignment_path = os.path.join('
|
46 |
frames = load_video(video_path)
|
47 |
print(frames.shape)
|
48 |
alignments = load_alignments(alignment_path)
|
|
|
41 |
file_name = path.split('/')[-1].split('.')[0]
|
42 |
# File name splitting for windows
|
43 |
file_name = path.split('\\')[-1].split('.')[0]
|
44 |
+
video_path = os.path.join('data','s1',f'{file_name}.mpg')
|
45 |
+
alignment_path = os.path.join('data','alignments','s1',f'{file_name}.align')
|
46 |
frames = load_video(video_path)
|
47 |
print(frames.shape)
|
48 |
alignments = load_alignments(alignment_path)
|