File size: 2,067 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 60 61 62 |
# Copyright (c) OpenMMLab. All rights reserved.
import argparse
import glob
import os
import os.path as osp
from multiprocessing import Pool
import mmengine
def extract_audio_wav(line):
"""Extract the audio wave from video streams using FFMPEG."""
video_id, _ = osp.splitext(osp.basename(line))
video_dir = osp.dirname(line)
video_rel_dir = osp.relpath(video_dir, args.root)
dst_dir = osp.join(args.dst_root, video_rel_dir)
os.popen(f'mkdir -p {dst_dir}')
try:
if osp.exists(f'{dst_dir}/{video_id}.wav'):
return
cmd = f'ffmpeg -i ./{line} -map 0:a -y {dst_dir}/{video_id}.wav'
os.popen(cmd)
except BaseException:
with open('extract_wav_err_file.txt', 'a+') as f:
f.write(f'{line}\n')
def parse_args():
parser = argparse.ArgumentParser(description='Extract audios')
parser.add_argument('root', type=str, help='source video directory')
parser.add_argument('dst_root', type=str, help='output audio directory')
parser.add_argument(
'--level', type=int, default=2, help='directory level of data')
parser.add_argument(
'--ext',
type=str,
default='mp4',
choices=['avi', 'mp4', 'webm'],
help='video file extensions')
parser.add_argument(
'--num-worker', type=int, default=8, help='number of workers')
args = parser.parse_args()
return args
if __name__ == '__main__':
args = parse_args()
mmengine.mkdir_or_exist(args.dst_root)
print('Reading videos from folder: ', args.root)
print('Extension of videos: ', args.ext)
fullpath_list = glob.glob(args.root + '/*' * args.level + '.' + args.ext)
done_fullpath_list = glob.glob(args.dst_root + '/*' * args.level + '.wav')
print('Total number of videos found: ', len(fullpath_list))
print('Total number of videos extracted finished: ',
len(done_fullpath_list))
pool = Pool(args.num_worker)
pool.map(extract_audio_wav, fullpath_list)
|