Python - Real-ESRGAN 提升图像、视频清晰度 - 最高可达 4 K
from realesrgan import RealESRGANer
# 实例化 RealESRGANer 对象,并指定模型为 ESRGAN (默认) 或 ESRGAN_x4 (4K 超分辨率)
model_name = 'ESRGAN' # 或者 'ESRGAN_x4'
realesrgan = RealESRGANer(model_path='Path to your model.pth', model=model_name)
# 提升单张图像的清晰度
input_image_path = 'Path to your input image'
output_image_path = 'Path to save the output image'
realesrgan(input_image_path, output_image_path, tile=2, scale=4) # tile 是图片分块处理,scale 是放大倍数
# 提升视频的清晰度
input_video_path = 'Path to your input video'
output_video_path = 'Path to save the output video'
realesrgan.enhance_video(input_video_path, output_video_path, scale=4)
# 注意:以上代码中的 'Path to your model.pth' 和 'Path to your input image/video' 需要替换为实际路径。
# tile 和 scale 参数可能需要根据您的具体需求和硬件性能进行调整。
以上代码示例展示了如何使用 Real-ESRGAN 提升单张图像和视频的清晰度。需要注意的是,您需要根据自己的环境配置正确的模型路径和输入输出路径。
评论已关闭