enhance.py

# multithreaded support 
# https://github.com/opencv/opencv/blob/master/samples/python/video_threaded.py

#Pyinstaller build command:
#   pyinstaller.exe "enhance.py" -F --add-data "src;src" --add-data "tensorflow_core;tensorflow_core"

"""
Scale up and improve quality of videos using Image Super Resolution
"""

__author__ = "Viraat Das"
__version__ = "0.1.0"
__license__ = "MIT"

import argparse
from src.generate_upscale import upscale_video

def main(args):
    # parameters from arg
    filename = args.filename
    remove_noise = args.remove_noise
    scale_factor = int(args.zoom)
    output_filename = args.output_filename
    
    # create new frames apply Super Resolution
    up_video = upscale_video(filename, remove_noise, scale_factor, output_filename)
    
    # creates directory with upscaled images from input file
    up_video.upscale_images_from_video()

    # combine images into video 
    up_video.combine_img_dir_into_video()

    # take outputted video and apply audio from input file to it
    up_video.extract_and_apply_audio()


if __name__ == "__main__":

    parser = argparse.ArgumentParser()

    parser.add_argument("filename", help="Required positional argument")
    parser.add_argument("--zoom", default="2", help="Specifies scaling factor of video")
    parser.add_argument("--remove_noise", action="store_true", help="If specified, then will denoise the video")
    parser.add_argument("--output_filename", default="output_video.mp4", help="Specify output filename")
    parser.add_argument(
        "--version",
        action="version",
        version="%(prog)s (version {version})".format(version=__version__))

    args = parser.parse_args()
    main(args)