tsm_gen_fine_label

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)