File size: 1,787 Bytes
d670799 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
# Copyright (c) OpenMMLab. All rights reserved.
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)
|