|
|
| import os
|
| import os.path as osp
|
| import subprocess
|
|
|
| import mmengine
|
|
|
| data_root = '../../../data/gym'
|
| video_root = f'{data_root}/videos'
|
| anno_root = f'{data_root}/annotations'
|
| anno_file = f'{anno_root}/annotation.json'
|
|
|
| event_anno_file = f'{anno_root}/event_annotation.json'
|
| event_root = f'{data_root}/events'
|
|
|
| videos = os.listdir(video_root)
|
| videos = set(videos)
|
| annotation = mmengine.load(anno_file)
|
| event_annotation = {}
|
|
|
| mmengine.mkdir_or_exist(event_root)
|
|
|
| for k, v in annotation.items():
|
| if k + '.mp4' not in videos:
|
| print(f'video {k} has not been downloaded')
|
| continue
|
|
|
| video_path = osp.join(video_root, k + '.mp4')
|
|
|
| for event_id, event_anno in v.items():
|
| timestamps = event_anno['timestamps'][0]
|
| start_time, end_time = timestamps
|
| event_name = k + '_' + event_id
|
|
|
| output_filename = event_name + '.mp4'
|
|
|
| command = [
|
| 'ffmpeg', '-i',
|
| '"%s"' % video_path, '-ss',
|
| str(start_time), '-t',
|
| str(end_time - start_time), '-c:v', 'libx264', '-c:a', 'copy',
|
| '-threads', '8', '-loglevel', 'panic',
|
| '"%s"' % osp.join(event_root, output_filename)
|
| ]
|
| command = ' '.join(command)
|
| try:
|
| subprocess.check_output(
|
| command, shell=True, stderr=subprocess.STDOUT)
|
| except subprocess.CalledProcessError:
|
| print(
|
| f'Trimming of the Event {event_name} of Video {k} Failed',
|
| flush=True)
|
|
|
| segments = event_anno['segments']
|
| if segments is not None:
|
| event_annotation[event_name] = segments
|
|
|
| mmengine.dump(event_annotation, event_anno_file)
|
|
|
