Ma/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2026-03-05 06:27:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactored interpolator

Browse Source
This commit is contained in:
k4yt3x
2023-05-25 18:44:54 +00:00
parent 5255e20283
commit cc01f2d8e2
2 changed files with 44 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
video2x/video2x.py
View File

@@ -69,7 +69,7 @@ from video2x.processor import Processor
from . import __version__
from .decoder import VideoDecoder, VideoDecoderThread
from .encoder import VideoEncoder
from .interpolator import Interpolator
from .interpolator import Interpolator, InterpolatorProcessor
from .upscaler import Upscaler, UpscalerProcessor
# for desktop environments only
@@ -102,7 +102,7 @@ class ProcessingSpeedColumn(ProgressColumn):
class ProcessingMode(Enum):
UPSCALE = {"label": "Upscaling", "processor": UpscalerProcessor}
INTERPOLATE = {"label": "Interpolating", "processor": Interpolator}
INTERPOLATE = {"label": "Interpolating", "processor": InterpolatorProcessor}
class Video2X:
@@ -179,7 +179,7 @@ class Video2X:
# elif mode == ProcessingMode.INTERPOLATE:
else:
standalone_processor: Any = Interpolator.ALGORITHM_CLASSES[
processing_settings[2]
processing_settings[1]
]
if getattr(standalone_processor, "process", None) is None:
standalone_processor().process_video(
@@ -222,7 +222,7 @@ class Video2X:
logger.info("Starting video encoder")
encoder = VideoEncoder(
input_path,
frame_rate * 2 if mode == "interpolate" else frame_rate,
frame_rate * 2 if mode == ProcessingMode.INTERPOLATE else frame_rate,
output_path,
output_width,
output_height,
@@ -337,6 +337,7 @@ class Video2X:
else:
logger.info("Processing has completed")
logger.info("Writing video trailer")
finally:
# stop keyboard listener