汤姆叔叔官方网站建设,公司网站建设佛山哪家好,wordpress抱歉您不能上传文件,建设网站建设哪家便宜目录
一.引言
二.服务构建
1.主函数 upload_gif
2.文件接收
3.专属目录
4.图像修复
5.gif2mp4
6.mp42gif
7.图像返回
三.服务测试
1.服务启动
2.服务调用
四.总结 一.引言
前面我们介绍了如何使用 Real-ESRGAN 进行图像增强并在原始格式 jpeg、jpg、mp4 的基础上…
目录
一.引言
二.服务构建
1.主函数 upload_gif
2.文件接收
3.专属目录
4.图像修复
5.gif2mp4
6.mp42gif
7.图像返回
三.服务测试
1.服务启动
2.服务调用
四.总结 一.引言
前面我们介绍了如何使用 Real-ESRGAN 进行图像增强并在原始格式 jpeg、jpg、mp4 的基础上增加了 gif 图像的处理
◆ 图像修复: Python - Real-ESRGAN 提升图像、视频清晰度
◆ 动图修复: Python - Real-ESRGAN 提高 gif 图像质量
为了方便随时随地上传图片即可修图我们将上面的内容结合通过 Flask 框架搭建图像修复服务。 二.服务构建
1.主函数 upload_gif
from flask import Flask, request, send_file, abort
from moviepy.editor import VideoFileClip
import os
import subprocess
import shutil
from enhance_frame import enhanceapp Flask(__name__)output./server/app.route(/upload-gif, methods[POST])
def upload_gif():if gif not in request.files:return abort(400, No gif file part in the request)file request.files[gif]# 确保文件名安全if file.filename :return abort(400, No selected file)if file and file.filename.lower().endswith(.gif):# 1.创建专属文件夹gif_name file.filename.split(.)[0]cur_dir os.path.join(output, gif_name)# 临时测试#shutil.rmtree(cur_dir)if not os.path.isdir(cur_dir):print(f文件夹 {cur_dir} 不存在创建中 ...)os.makedirs(cur_dir)# 2.图像转存至本地并切换为 mp4 格式input_path os.path.join(cur_dir, file.filename)print(f原始图像保存路径 {input_path})file.save(input_path)mp4_path, w, h gif2mp4(input_path, gif_name, cur_dir)print(fGIF 转 MP4 路径 {mp4_path})# 3.对图像进行增强行理enhance(mp4_path, cur_dir)enhance_output os.path.join(cur_dir, gif_name _trans_out.mp4)# 4.转换为 gif 并返回final_output os.path.join(cur_dir, gif_name _out.gif)mp42gif(enhance_output, final_output, w, h)else:print(f文件夹 {cur_dir} 存在 ...)final_output uploaded.gifreturn send_file(final_output, mimetypeimage/gif)return abort(400, The uploaded file is not a GIF)
主函数为 Post 形式其通过 request.files 接收传入的 gif 图像并对图像进行修复返回修复后的 gif 图像。下面分步骤对修复过程中的代码进行讲解。 2.文件接收 if gif not in request.files:return abort(400, No gif file part in the request)file request.files[gif]# 确保文件名安全if file.filename :return abort(400, No selected file)
首先检查 gif 是否在传入的图像中随后通过 .files[gif] 获取传入的 gif 图像最后检查文件名为空的异常情况。这里只做了简单的异常判断大家可以根据自己场景进行扩充。 3.专属目录 if file and file.filename.lower().endswith(.gif):# 1.创建专属文件夹gif_name file.filename.split(.)[0]cur_dir os.path.join(output, gif_name)# 临时测试#shutil.rmtree(cur_dir)if not os.path.isdir(cur_dir):print(f文件夹 {cur_dir} 不存在创建中 ...)os.makedirs(cur_dir)
这里通过 file.filename 获取对应 gif 图的名称并在输出文件中创建该 gif 图的专属文件夹一方面是可以记录 Post 请求传来的图像以及修复后的状态另一方面相当于做了一个本地 Cache如果相同名称的 gif 图像再次 Post 传过来可以直接在文件夹中找到缓存的返回。 4.图像修复 # 2.图像转存至本地并切换为 mp4 格式input_path os.path.join(cur_dir, file.filename)print(f原始图像保存路径 {input_path})file.save(input_path)mp4_path, w, h gif2mp4(input_path, gif_name, cur_dir)print(fGIF 转 MP4 路径 {mp4_path})# 3.对图像进行增强行理enhance(mp4_path, cur_dir)enhance_output os.path.join(cur_dir, gif_name _trans_out.mp4)# 4.转换为 gif 并返回final_output os.path.join(cur_dir, gif_name _out.gif)mp42gif(enhance_output, final_output, w, h)
由于原始的 Real-ESRGAN 源码对视频只支持了 flv 和 mp4因此对于要修复的 gif 我们需要
◆ gif2mp4
先将其逐帧转换为 mp4 这一步由 gif2mp4 函数负责
◆ enhance
转换为 mp4 后由 enhance 函数进行修复这里 enhance 函数对应源码中的 inference_realesrgan_video.py enhance 函数对应源码中的 main() 函数只需要修改名称即可
◆ mp42gif
修复后再将 mp4 转换为 gif这一步由 mp42gif 负责。
Tips
如果觉得上述过程比较繁琐可以直接通过 ffmpeg 在源码中实现对 gif 图像进行转换。 5.gif2mp4
def gif2mp4(file_path, filename, out):# 加载GIF文件clip VideoFileClip(file_path)# 输出地址output os.path.join(out, filename _trans.mp4)# 将GIF文件写到MP4文件clip.write_videofile(output, fpsclip.fps, codeclibx264, bitrate8000k)# 关闭clip.close()return output, clip.w, clip.h
通过 moviepy 的 VideoFileClip 函数读取 mp4随后通过 write_videofile 将 mp4 写出这里 bitrate 可以控制输出 mp4 的码率数值越高可以在一定程度提高输出视频的质量。同时为了保证输出的 gif 图像与原始 gif 图像的长宽保持一致这里我们还返回了 clip.w 和 clip.h。 6.mp42gif
def mp42gif(file_path, out, w, h):# 加载GIF文件clip VideoFileClip(file_path)# 目标大小resized_clip clip.resize(widthw).resize(heighth)# 导出GIFresized_clip.write_gif(out, fps15) # fps参数可以减少帧数来进一步降低文件大小# 释放资源clip.close()
还是通过 moviepy 进行 mp4 到 gif 的转换使用 resize 保证输出的 gif 与原始 gif 尺寸相同由于增强的缘故新生成的 gif 会比之前的老图大 5-10 倍如果有缩减的需要可以减少 FPS 或者使用 reduction_factor 按百分比减少分辨率。 7.图像返回 else:print(f文件夹 {cur_dir} 存在 ...)final_output uploaded.gifreturn send_file(final_output, mimetypeimage/gif)return abort(400, The uploaded file is not a GIF)
else 逻辑其实是该 gif 存在缓存文件夹直接在文件夹找增强过的图像直接返回就可以了这里省略了 Cache 的逻辑有需要的同学也可以自己添加。最后就是通过 send_file 将 final_output 文件写出作为 Post 的结果回传。 三.服务测试
1.服务启动
if __name__ __main__:# 这将允许你在你的开发机上本地运行它ip 10.196.1.666app.run(hostip, port5000, debugTrue)
运行主函数后提示下述信息即代表 Http 服务启动Running on http: 后面就是对应的 URL。如果显示异常可以尝试切换 port 重新启动 2.服务调用
在本地执行下述命令:
#!/bin/bash# gif 路径
gif_pathgif/Users/xxx/Desktop/892f4f60de074a7b9c682f744a5cf6f8.gif
# post 调用 URL
urlhttp://10.192.1.666:5000/upload-gif
# 输出地址
outresponse.gifcurl -X POST $url -F $gif_path --output $out
◆ 服务端日志 服务端日志会把 Flask 程序里相关的 print 都打印出来最后 inference 是视频修复的 tqdm 进度条这里 nb_frames: 12 代表当前 gif 转换的 mp4 共包含 12 帧程序在单线程的情况下需要逐帧转换有需要的同学也可以使用多线程实现分 Frame 处理最后使用 ffmpeg 进行合并。 ◆ 本地日志 本地日志会记录调用的时间以及相关接收数据等信息。 ◆ gif 修复前 ◆ gif 修复后 四.总结
本文使用 Flask 搭建了轻量级的图像修复服务可以优化的点还有很多大家可以多多交流讨论。
