阳江市建设路龙源学校网站,广西壮族自治区建设厅官方网站,中国空间站建造完成,做3d效果在哪个网站类似的图片修复项目 GFPGAN 的改造见我另一篇文 https://blog.csdn.net/weixin_43074462/article/details/132497146 DeOldify 是一款开源软件#xff0c;用于给黑白照片或视频上色#xff0c;效果还不错。
安装部署教程请参考别的文章#xff0c;本文基于你给项目跑通用于给黑白照片或视频上色效果还不错。
安装部署教程请参考别的文章本文基于你给项目跑通并 pip 集成过 flask 后
本修改支持加载一次模型持续提供 HTTP 服务
在根目录新建主文件 run_serv.py
import iofrom deoldify import device
from deoldify.device_id import DeviceId
from deoldify.visualize import *
from flask import Flask, request, Responseapp Flask(__name__)# 提取通用代码
# choices: CPU, GPU0...GPU7
device.set(deviceDeviceId.GPU0)
plt.style.use(dark_background)
colorizer get_image_colorizer(artisticTrue)
torch.backends.cudnn.benchmark Trueapp.route(/action, methods[POST, GET])
def action():file request.files[file]img_name file.filename_, ext os.path.splitext(img_name)img_in_memory io.BytesIO(file.read())#此处的35应该是种子而不是参数 可以改成25或者什么的影响似乎不大img_ret colorizer.get_transformed_image_pzj(img_in_memory, 35)# colorizer.plot_transformed_image(pathC:/Users/PiaoZhenJia/Pictures/微信图片_20230728194252.png,# render_factor35, results_dirPath(D:/), post_processFalse, watermarkedFalse)img_ret_bytes io.BytesIO()img_ret.save(img_ret_bytes, formatPNG)img_ret_bytes.seek(0)res Response(img_ret_bytes)res.headers.add(Content-Type, image/ ext)res.headers.add(Content-Disposition, fattachment; filenamecolorize_{img_name})return resdef main():app.run(port2021, host127.0.0.1, debugFalse)if __name__ __main__:main()
然后修改 visualize.py 我选择在187行附近加入这个函数 def get_transformed_image_pzj(self, img_bytes: io.BytesIO, render_factor: int None) - Image:self._clean_mem()orig_image PIL.Image.open(img_bytes)filtered_image self.filter.filter(orig_image, orig_image, render_factorrender_factor)return filtered_image测试方法如图 接受一个form表单file字段传输二进制图片文件返回结果直接为二进制图片。可以通过 PostMan 的 SaveResponse 保存到一个 PNG 文件
