|
|
|
|
|
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)
|
|
|
|
