import re
import os
label_file = "ucfTrainTestlist/classInd.txt"
input_files = ["/dataset/ucfTrainTestlist/testlist01.txt"]
input_path_prefix = "/dataset/UCF101_frames"
class_ind = [x.strip().split() for x in open(label_file)] #label
dict_categories = {x[1]:int(x[0])-1 for x in class_ind}
filename_output = "rgb_val_split_1.txt"
for input_file_path in input_files:
with open(input_file_path) as f:
for line in f.readlines():
line = line.strip()
match = re.split(r'/|\.', line)
category = match[0]
frame_folder = match[1]
category_index = dict_categories[category]
frame_count = len(os.listdir(os.path.join(input_path_prefix, category, frame_folder)))-1
output=(f'{category}/{frame_folder} {frame_count} {category_index}\n')
print(output)
with open(filename_output, 'a') as f:
f.writelines(output)