diff --git a/src/datasources/common.py b/src/datasources/common.py index a03d785..721f291 100644 --- a/src/datasources/common.py +++ b/src/datasources/common.py @@ -57,7 +57,7 @@ def __init__(self, video_path, frame_indices=None, is_async=True, output_size=No if self.video_path.endswith('_eyes.mp4'): self.timestamps_path = video_path.replace('_eyes.mp4', '.timestamps.txt') elif self.video_path.endswith('_face.mp4'): - self.timestamps_path = video_path.replace('_eyes.mp4', '.timestamps.txt') + self.timestamps_path = video_path.replace('_face.mp4', '.timestamps.txt') elif self.video_path.endswith('.128x72.mp4'): self.timestamps_path = video_path.replace('.128x72.mp4', '.timestamps.txt') else: