| | """
|
| | Script for extracting audio (16-bit, mono, 22000 Hz) from video file.
|
| | """
|
| |
|
| | import os
|
| | import argparse
|
| | import subprocess
|
| |
|
| |
|
| | def parse_args():
|
| | """
|
| | Create python script parameters.
|
| |
|
| | Returns
|
| | -------
|
| | ArgumentParser
|
| | Resulted args.
|
| | """
|
| | parser = argparse.ArgumentParser(
|
| | description="Extract audio from video file",
|
| | formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
| | parser.add_argument(
|
| | "--in-video",
|
| | type=str,
|
| | required=True,
|
| | help="path to input video file or directory")
|
| | parser.add_argument(
|
| | "--out-audio",
|
| | type=str,
|
| | help="path to output audio file")
|
| |
|
| | args = parser.parse_args()
|
| | return args
|
| |
|
| |
|
| | def extract_audio(in_video,
|
| | out_audio):
|
| | """
|
| | Real extract audio from video file.
|
| |
|
| | Parameters
|
| | ----------
|
| | in_video : str
|
| | Path to input video file.
|
| | out_audio : str
|
| | Path to output audio file.
|
| | """
|
| | if not out_audio:
|
| | file_stem, _ = os.path.splitext(in_video)
|
| | out_audio = file_stem + ".wav"
|
| |
|
| |
|
| |
|
| | command = "ffmpeg -i {in_video} -vn -acodec pcm_s16le -ac 1 -ar 16000 {out_audio}"
|
| | subprocess.call([command.format(in_video=in_video, out_audio=out_audio)], shell=True)
|
| |
|
| |
|
| | def main():
|
| | """
|
| | Main body of script.
|
| | """
|
| | args = parse_args()
|
| | in_video = os.path.expanduser(args.in_video)
|
| | if not os.path.exists(in_video):
|
| | raise Exception("Input file/directory doesn't exist: {}".format(in_video))
|
| | if os.path.isfile(in_video):
|
| | extract_audio(
|
| | in_video=in_video,
|
| | out_audio=args.out_audio)
|
| | else:
|
| | video_file_paths = []
|
| | for file_name in os.listdir(in_video):
|
| | if not os.path.isfile(os.path.join(in_video, file_name)):
|
| | continue
|
| | _, file_ext = os.path.splitext(file_name)
|
| | if file_ext.lower() in (".mp4", ".mkv", ".avi"):
|
| | video_file_path = os.path.join(in_video, file_name)
|
| | video_file_paths.append(video_file_path)
|
| | video_file_paths = sorted(video_file_paths)
|
| | for video_file_path in video_file_paths:
|
| | extract_audio(
|
| | in_video=video_file_path,
|
| | out_audio="")
|
| |
|
| |
|
| | if __name__ == "__main__":
|
| | main()
|
| |
|
