def show_upscaled_image(
source_path: pathlib.Path,
parameters: Parameters = Parameters(),
GPU_mode: bool = False,
ACNet: bool = True,
):
""" display an image processed by Anime4K09 or ACNet
Args:
source_path: input file path.
parameters (Parameters, optional): custom arguments passed to Anime4KCPP.
GPU_mode (bool, optional): enable GPU mode. Defaults to False.
ACNet (bool, optional): enable ACNet mode. Defaults to True.
Raises:
ACError
"""
if GPU_mode:
if ACNet:
ac_object = AC(
False, True, type=ProcessorType.GPUCNN, parameters=parameters
)
else:
ac_object = AC(True, False, type=ProcessorType.GPU, parameters=parameters)
else:
if ACNet:
ac_object = AC(
False, False, type=ProcessorType.CPUCNN, parameters=parameters
)
else:
ac_object = AC(False, False, type=ProcessorType.CPU, parameters=parameters)
ac_object.load_image(str(source_path))
ac_object.process()
ac_object.show_image()
def show_upscaled_image(
source_path: pathlib.Path,
parameters: Parameters = Parameters(),
GPU_mode: bool = False,
ACNet: bool = True,
):
"""display an image processed by Anime4K09 or ACNet
Args:
source_path: input file path.
parameters (Parameters, optional): custom arguments passed to Anime4KCPP.
GPU_mode (bool, optional): enable GPU mode. Defaults to False.
ACNet (bool, optional): enable ACNet mode. Defaults to True.
Raises:
ACError
"""
if GPU_mode:
if ACNet:
ac_object = AC(
managerList=ac.ManagerList(
[ac.OpenCLACNetManager(pID=0, dID=0)]),
parameters=parameters,
type=ac.ProcessorType.OpenCL_ACNet,
)
else:
ac_object = AC(
managerList=ac.ManagerList(
[ac.OpenCLACNetManager(pID=0, dID=0)]),
parameters=parameters,
type=ac.ProcessorType.OpenCL_Anime4K09,
)
else:
if ACNet:
ac_object = AC(
managerList=None,
parameters=parameters,
type=ac.ProcessorType.CPU_ACNet,
)
else:
ac_object = AC(
managerList=None,
parameters=parameters,
type=ac.ProcessorType.CPU_Anime4K09,
)
ac_object.load_image(str(source_path))
ac_object.process()
ac_object.show_image()
def upscale_videos(
input_paths: list,
output_suffix: str = "_output",
output_path: pathlib.Path = None,
parameters: Parameters = Parameters(),
GPU_mode: bool = False,
ACNet: bool = True,
codec: Codec = Codec.MP4V,
):
""" upscale a list of video files with Anime4k
Args:
input_paths (list): list of input file paths
output_suffix (str, optional): output files suffix. Defaults to "_output".
output_path (pathlib.Path, optional): parent directory of output paths. Defaults to None.
parameters (Parameters, optional): custom arguments passed to Anime4KCPP.
GPU_mode (bool, optional): enable GPU mode. Defaults to False.
ACNet (bool, optional): enable ACNet mode. Defaults to True.
codec (Codec, optional): codec for video encodeing. Defaults to MP4V
Raises:
FileExistsError: when output path exists and isn't a directory
ACError
"""
# sanitize input list
input_paths = _sanitize_input_paths(input_paths)
# if destination path unspecified
if output_path is None:
# destination path is first input file's parent directory
output_path = input_paths[0].parent
# if destination path doesn't exist
if not output_path.exists():
# create directory and its parents if necessary
output_path.mkdir(parents=True, exist_ok=True)
# else if it already exists but isn't a directory
elif not output_path.is_dir():
raise FileExistsError(
"destination path already exists and isn't a directory")
# set parameters to video mode
parameters.videoMode = True
# create anime4k object
if GPU_mode:
if ACNet:
ac_object = AC(False,
True,
type=ProcessorType.GPUCNN,
parameters=parameters)
else:
ac_object = AC(True,
False,
type=ProcessorType.GPU,
parameters=parameters)
else:
if ACNet:
ac_object = AC(False,
False,
type=ProcessorType.CPUCNN,
parameters=parameters)
else:
ac_object = AC(False,
False,
type=ProcessorType.CPU,
parameters=parameters)
# process each of the files in the list
for path in input_paths:
# create temporary directory to save the upscaled video
temporary_directory = pathlib.Path(tempfile.mkdtemp())
temporary_video_file_path = temporary_directory.joinpath("temp.mp4")
# process and save video file to temp/temp.mp4
ac_object.load_video(str(path))
ac_object.set_save_video_info(str(temporary_video_file_path), codec)
ac_object.process_with_progress()
ac_object.save_video()
ffmpeg_handler.migrate_audio_streams(
upscaled_video=temporary_video_file_path,
original_video=path,
output_path=(output_path.joinpath(path.stem + output_suffix +
path.suffix)),
)
def upscale_images(
input_paths: list,
output_suffix: str = "_output",
output_path: pathlib.Path = None,
parameters: Parameters = Parameters(),
GPU_mode: bool = False,
ACNet: bool = True,
):
""" upscale a list of image files with Anime4K
Args:
input_paths (list): list of input file paths
output_suffix (str, optional): output files. Defaults to "_output".
output_path (pathlib.Path, optional): parent directory of output paths. Defaults to None.
parameters (Parameters, optional): custom arguments passed to Anime4KCPP.
GPU_mode (bool, optional): enable GPU mode. Defaults to False.
ACNet (bool, optional): enable ACNet mode. Defaults to True.
Raises:
FileExistsError: when output path exists and isn't a directory
ACError
"""
# sanitize input list
input_paths = _sanitize_input_paths(input_paths)
# if destination path unspecified
if output_path is None:
# destination path is first input file's parent directory
output_path = input_paths[0].parent
# if destination path doesn't exist
if not output_path.exists():
# create directory and its parents if necessary
output_path.mkdir(parents=True, exist_ok=True)
# else if it already exists but isn't a directory
elif not output_path.is_dir():
raise FileExistsError(
"destination path already exists and isn't a directory")
# create Anime4K object
if GPU_mode:
if ACNet:
ac_object = AC(False,
True,
type=ProcessorType.GPUCNN,
parameters=parameters)
else:
ac_object = AC(True,
False,
type=ProcessorType.GPU,
parameters=parameters)
else:
if ACNet:
ac_object = AC(False,
False,
type=ProcessorType.CPUCNN,
parameters=parameters)
else:
ac_object = AC(False,
False,
type=ProcessorType.CPU,
parameters=parameters)
# process each of the files in the list
for path in input_paths:
# anime4k load and process image
ac_object.load_image(str(path))
ac_object.process()
# construct destination file path object
output_file_path = output_path.joinpath(
(path.stem + output_suffix + path.suffix))
print(f"Saving file to: {output_file_path}")
ac_object.save_image(str(output_file_path))