图片视频画质优化
Go to file
2019-03-09 12:55:27 -05:00
bin Merge remote-tracking branch 'origin/master' 2019-03-09 12:51:49 -05:00
.gitignore added gitignore line for pycharm 2018-11-02 21:48:52 -04:00
LICENSE Initial commit 2018-02-24 11:34:41 -05:00
README.md added important description for command line usage 2019-03-09 12:55:27 -05:00

Video2X Lossless Video Enlarger

Official Discussion Group (Telegram): https://t.me/video2x

Prerequisites

Component names that are bolded are mandatory.

Component names that are italicized can be automatically downloaded and configured with the video2x_setup.py script.

  1. Operating System: Windows
  2. AMD GPU / Nvidia GPU
  3. AMD GPU driver / Nvidia GPU driver / Nvidia CUDNN
  4. FFMPEG
  5. waifu2x-caffe / waifu2x-converter-cpp

Recent Changes

2.6.0 (March 9, 2019)

  • Complete redesign of configuration file format. The configuration file is now much more flexible and easy to look at.
  • Various modifications done to the rest of the program to adapt to the changes made in the configuration file. This eliminated some problems existed in the previous version.

2.5.0 (March 4, 2019)

  • Added progress bar according to @ArmandBernard 's suggestion.
  • Cache folders are now created under '{}\\video2x'.format(tempfile.gettempdir()) by default, and the default key for cache folder location in configuration file has been changed to video2x_cache_folder. This makes it easier to manage cache folders.
  • Cache folders are now more likely to be deleted successfully even if the program crashes. This is made possible with more try-catch blocks.
  • Updated FFMPEG stream migration grammar according to @cr08 's suggestions. All streams (including attachment streams) will now be copied over to the new video.
  • A lot of minor upgrades and adjustments.

2.4.3 (February 26, 2019)

  • Fixed the bug where ffmpeg arguments (hwaccel) are not passed on to ffmpeg.

2.4.2 (February 26, 2019)

  • Added the function to detect insufficient GPU memory.
  • Added the function to create temporary folders automatically.
  • Fixed configuration file path error.

Description

Video2X is an automation software based on waifu2x image enlarging engine. It extracts frames from a video, enlarge it by a number of times without losing any details or quality, keeping lines smooth and edges sharp.

For short: Video2X enlarges your video without losing details

Watch for the sharper edges in this screenshot around the shadows:

preview

You can also watch the YouTube video Demo: https://www.youtube.com/watch?v=PG94iPoeoZk

Clip is from trailer of animated movie "千と千尋の神隠し". Copyright belongs to "株式会社スタジオジブリ (STUDIO GHIBLI INC.)". Will delete immediately if use of clip is in violation of copyright.

Screenshot

screenshot

Installation

Prerequisites

Install Dependencies

First, clone the video2x repository.

$ git clone https://github.com/K4YT3X/video2x.git
$ cd video2x/bin

Then you may run the video2x_setup.py script to install and configure the depencies automatically. This script is designed and tested on Windows 10.

This script will install ffmpeg, waifu2x-caffe to %LOCALAPPDATA%\\video2x and all python libraries.

$ python video2x_setup.py

Alternatively, you can also install the dependencies manually. Please refer to the prerequisites section to see what's needed.

Then you'll need to install python dependencies before start using video2x. Install simply by executing the following command.

$ pip install -r requirements.txt

Quick Start

Sample Videos

If you can't find a video clip to begin with, or if you want to see a before-after comparison, we have prepared some sample clips for you. The quick start guide down below will also be based on the name of the sample clips.

sample_video

Clip is from anime "さくら荘のペットな彼女". Copyright belongs to "株式会社アニプレックス (Aniplex Inc.)". Will delete immediately if use of clip is in violation of copyright.

For Command Line Beginners

If you're unfamiliar of directories in command lines, then here's a short section that might help you to get started.

For example, if you downloaded the sample input video to C:\Users\[YourUsername]\Downloads, then the full path of your input video will be C:\Users\[YourUsername]\Downloads\sample_input.mp4, vice versa. The output path is also relative. If you want to export the output video to the current directory, just specify the output video name such as output.mp4. However, if you want to put the output video in a different directory, you should use relative or absolute path, such as C:\Users\[YourUsername]\Desktop\output.mp4.

If you're tired typing everything in, you can also drag the video file directly into the command line window, and Windows will fill in the full path of the video for you.

Note that all command line arguments/options overwrite configuration file settings.

Nvidia CUDA (waifu2x-caffe)

Enlarge the video to 1920x1080 using CUDA. You may also use the -r/--ratio option.

$ python video2x.py -i sample_input.mp4 -o sample_output.mp4 -m gpu --width=1920 --height=1080

Corresponding sample configuration:

"waifu2x_caffe": {
  "waifu2x_caffe_path": "C:\\Users\\K4YT3X\\AppData\\Local\\video2x\\waifu2x-caffe\\waifu2x-caffe-cui.exe",
  "mode": "noise_scale",
  "scale_ratio": null,
  "scale_width": null,
  "scale_height": null,
  "noise_level": 3,
  "process": "gpu",
  "crop_size": 128,
  "output_quality": -1,
  "output_depth": 8,
  "batch_size": 1,
  "gpu": 0,
  "tta": 0,
  "input_path": null,
  "output_path": null,
  "model_dir": "models/cunet",
  "crop_w": null,
  "crop_h": null
}

Nvidia CNDNN

Enlarge the video to 1920x1080 using CUDNN. You may also use the -r/--ratio option.

$ python video2x.py -i sample_input.mp4 -o sample_output.mp4 -m cudnn --width=1920 --height=1080

AMD or Nvidia (waifu2x-converter-cpp OpenCL)

Enlarge the video by 2 times using OpenCL. Note that waifu2x-converter-cpp doesn't support width and height. You'll also have to explicitly specify that the driver to be used is waifu2x_converter.

$ python video2x.py -i sample_input.mp4 -o sample_output.mp4 -m gpu -r 2 -d waifu2x_converter

Corresponding sample configuration is shown below. Note that the waifu2x_path is different from the configuration for waifu2x-caffe. Instead of the binary path, folder containing extracted waifu2x-converter-cpp.exe should be specified.

"waifu2x_converter": {
  "waifu2x_converter_path": "C:\\Users\\K4YT3X\\AppData\\Local\\video2x\\waifu2x-converter-cpp",
  "block_size": null,
  "disable-gpu": null,
  "force-OpenCL": null,
  "processor": null,
  "jobs": null,
  "model_dir": null,
  "scale_ratio": null,
  "noise_level": 3,
  "mode": "noise_scale",
  "quiet": true,
  "output": null,
  "input": null
}

CPU

Enlarge the video to 1920x1080 using the CPU. You may also use the -r/--ratio option. This is potentially much slower than using a GPU. The configuration file for this method is similar to the previous methods.

$ python video2x.py -i sample_input.mp4 -o sample_output.mp4 -m cpu --width=1920 --height=1080

Configuration File Format

Video2X converts the configuration file keys and values directly into options or arguments. For example, in the following sample config, --scale_width 1920 will be passed to waifu2x.

"waifu2x_caffe": {
  ...
  "scale_width": null,
  ...
}
  • Keys having a value of null (or false, not recommended as it can mean different things in the future) means to ignore this key, thus ignoring both the key and the value.
  • Keys having a value of true means that this is an option. Only the key will be passed to waifu2x, not the value (e.g. --quiet).

Full Usage

General Options

-h, --help

show this help message and exit

File Input and Output

-i INPUT, --input INPUT

Specify source video file/directory

-o OUTPUT, --output OUTPUT

Specify output video file/directory

Upscaler Options

-m {cpu,gpu,cudnn}, --method {cpu,gpu,cudnn}

Specify upscaling method

-d {waifu2x_caffe,waifu2x_converter}, --driver {waifu2x_caffe,waifu2x_converter}

Waifu2x driver

-y {upconv_7_anime_style_art_rgb,upconv_7_photo,anime_style_art_rgb,photo,anime_style_art_y}, --model_type {upconv_7_anime_style_art_rgb,upconv_7_photo,anime_style_art_rgb,photo,anime_style_art_y}

Specify model to use

-t THREADS, --threads THREADS

Specify number of threads to use for upscaling

-c CONFIG, --config CONFIG

Manually specify config file

--width WIDTH

Output video width

--height HEIGHT

Output video height

-r RATIO, --ratio RATIO

Scaling ratio

License

Licensed under the GNU General Public License Version 3 (GNU GPL v3) https://www.gnu.org/licenses/gpl-3.0.txt

GPLv3 Icon

(C) 2018-2019 K4YT3X

Credits

This project relies on the following software and projects.