import util FRAME_PATH = "/media/data/mtriet/dataset/scnn_%s_frames" % sys.argv[1] SUB_PATH = "/media/data/mtriet/raw_video/%s/train" % (sys.argv[1]) WINDOW_SIZE = [16, 32, 64, 128, 256, 512] OVERLAP_RATE = 0.75 CLASSES = [[], []] # 0, 1 if len(sys.argv) < 3: print('fb pad=True/False') sys.exit(1) for window_size in WINDOW_SIZE: for frame_root, sub_folder, sub_files in os.walk(FRAME_PATH): for folder in sub_folder: subtitles = Subtitle.load_subtitle(SUB_PATH, folder, sys.argv[2]) frames = sorted(os.listdir(frame_root + '/' + folder)) sub_index = 0 for begin_pivot in range( 1, len(frames) - window_size, int(window_size * (1 - OVERLAP_RATE))): # ignore last few frames if (begin_pivot > subtitles[sub_index].end) and ( sub_index < len(subtitles) - 1): sub_index += 1 end_pivot = min(begin_pivot + window_size, len(frames)) segment = range(begin_pivot, begin_pivot + window_size + 1) sub_range = subtitles[sub_index].get_range() intersection = np.intersect1d(segment, sub_range)
import sys import os from subtitle import Subtitle SUB_PATH = "/media/data/mtriet/raw_video/%s/train" % (sys.argv[1]) FRAME_PATH = "/media/data/mtriet/dataset/scnn_%s_frames" % sys.argv[1] if __name__ == '__main__': if len(sys.argv) > 4 or len(sys.argv) < 3: print('command fb/bb front_pad rear_pad') sys.exit(0) for frame_root, sub_folder, _ in os.walk(FRAME_PATH): for folder in sub_folder: print(folder) subtitles = Subtitle.load_subtitle(SUB_PATH, folder, False) with open("%s/%s_pad.aqt" % (SUB_PATH, folder), 'w') as f: for s in subtitles: f.write(s.to_string(sys.argv[2], sys.argv[3]))