Python + AI人工智能——给老照片恢复色彩、上色
哈喽,大家好,今天给大家分享有趣的 AI 项目。
今天我们分享用 NoGAN
的图像增强技术给老照片着色。 喜欢记得收藏、关注、点赞。
注:资料数据
图片腾讯文档-在线文档https://docs.qq.com/doc/DU2NjQnZZZVhiZUxp
效果如下:
原图
上色后
NoGAN
是一种新型 GAN
,它能花费最少的时间进行 GAN
训练。
今天分享的这个项目已在 GitHub
开源项目,下面我们来运行一下。
1. 准备工作
首先,用 git clone
命令下载源码
git clone https://github.com/jantic/DeOldify.git
进入项目根目录,安装 Python
依赖包
pip3 install -r requirements.txt
编写代码运行项目之前,需要下载预训练好的模型。项目提供了三个模型
模型
区别如下:
ColorizeArtistic_gen.pth:在有趣的细节和活力方面实现了最高质量的图像着色效果,该模型在 UNet 上使用 resnet34 为主干,通过 NoGAN 进行了 5 次评论家预训练/GAN 循环重复训练
ColorizeStable_gen.pth:在风景和肖像方面取得了最佳效果,该模型在 UNet 上使用 resnet101 为主干,通过 NoGAN 进行了 3 次评论家预训练/GAN 循环重复训练
ColorizeVideo_gen.pth:针对流畅的视频进行了优化,它仅使用初始生成器/评论家预训练/GAN NoGAN 训练。由于追求流畅的速度,它的色彩比前两者少。
将下载好的模型文件放在项目根目录的 models
目录下即可。
2. 编写代码
在项目根目录同级目录下创建 Python
文件,编写代码加载刚刚下载好的模型文件。
from DeOldify.deoldify.generators import gen_inference_wide from DeOldify.deoldify.filters import MasterFilter, ColorizerFilter# 指定模型文件 learn = gen_inference_wide(root_folder=Path('./DeOldify'), weights_name='ColorizeVideo_gen')# 加载模型 deoldfly_model = MasterFilter([ColorizerFilter(learn=learn)], render_factor=10)
root_folder
指定项目根目录, weights_name
指定接下来使用哪个模型为照片上色。
读取老照片,进行上色
import cv2 import numpy as np from PIL import Imageimg = cv2.imread('./images/origin.jpg') img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) pil_img = Image.fromarray(img)filtered_image = deoldfly_model.filter(pil_img, pil_img, render_factor=35, post_process=True )result_img = np.asarray(filtered_image) result_img = cv2.cvtColor(result_img, cv2.COLOR_RGB2BGR) cv2.imwrite('deoldify.jpg', result_img)
用 cv2
读取老照片,并用 PIL.Image
模块将图片转换成模型输入所需要的格式,送入模型进行上色,完成后保存。
上述代码是我从项目源码中抽取的,可以看到,运行代码还是非常简单的。
感兴趣的朋友可以自己运行一下,也可以试试其他模型的效果。
Python + AI人工智能——给老照片恢复色彩、上色相关推荐
- python注册人工智能专业_从专业程度上分析Python和人工智能(AI) 它们如何相关?...
Python和人工智能(AI) - 它们如何相关? Python是当今开发人员使用的最流行的编程语言之一.Guido Van Rossum于1991年创建它,自成立以来,它一直是使用最广泛的语言之一, ...
- 四天人工智能 python入门体验课_百度深度学习7天打卡营,用Python+AI识别“青你2”小姐姐的高颜值...
原标题:百度深度学习7天打卡营,用Python+AI识别"青你2"小姐姐的高颜值 "淡黄的长裙,蓬松的头发",一夜之间洗脑全网,小姐姐们实在太让人上头了! 导师 ...
- python在人工智能应用锁_python实现人工智能Ai抠图功能
自己是个PS小白,没办法只能通过技术来证明自己. 话不多说,直接上代码 from removebg import RemoveBg import requests import os if __nam ...
- Java和Python,哪个更适合开发AI人工智能?
当今的机器世界需要能够改变我们生活.工作和娱乐方式的技术.考虑到智能系统和具有行为算法.智能搜索和能够自行学习的智能系统的需求,约翰麦卡锡将人工智能引入了我们的世界,并被亲切地称为人工智能. 这些 A ...
- 五节课从零起步(无需数学和Python基础)编码实现AI人工智能框架电子书V1
五节课从零起步 (无需数学和Python 基础) 编码实现AI 人工智能框架 王 家 林 2018/4/15 ...
- AI老照片自动上色,复现记忆中的旧时光?
我们知道深度学习已经应用到几乎所有领域,但是如果我们能够建立一个基于深度学习的模型,可以为旧照片上色并重现我们童年的旧记忆,那该是多么令人兴奋啊!那么我们该怎么办? 今天将为大家介绍一个通过深度学习的 ...
- AI人工智能 Python实现人机对话
这篇文章主要为大家详细介绍了AI人工智能应用,本文拟使用Python开发语言实现类似于WIndows平台的"小娜",,具有一定的参考价值,感兴趣的小伙伴们可以参考一下 在人工智能进 ...
- python图片自动上色_通过深度学习方法为黑白老照片自动上色,带我们重新忆起那段老时光!...
我们知道,深度学习几乎已经应用在每一个领域,但如果我们能够构建一个基于深度学习的模型,让它能够给老照片着色,重现我们童年的旧回忆,这该多么令人激动啊!那么我们要怎么做呢?本文的作者将为大家介绍一个教程 ...
- 《Python从入门到放弃》(Yanlz+Unity+SteamVR+云计算+5G+AI=VR云游戏=Python+PyCharm+人工智能+无人驾驶+数据可视化+人机交互+立钻哥哥+==)
<Python从入门到放弃> <Python从入门到放弃> 版本 作者 参与者 完成日期 备注 YanlzAI_Python_V01_1.0 严立钻 2019.09.25 ## ...
最新文章
- 面试使用计算机,面试相关之计算机基础
- Docker系列 四.Docker部署SpringBoot
- 简述python的安装过程_python3+ selenium3开发环境搭建-手把手教你安装python(详细)...
- Eureka自我保护机制
- 前端开发需要掌握什么技术,成为一个高薪的web前端程序员很难吗?
- 攻击者巧妙滥用谷歌 DoH 下载恶意软件
- 搭建etcd 3.4.15集群(详细教程,包括选举过程、数据备份和恢复)
- Windows下 Anaconda + VScode Python 环境搭建 多图 非常详细
- BZOJ4011:[HNOI2015]落忆枫音(DP,拓扑排序)
- Matlab相关工具箱下载地址汇总
- C#中LitJson的使用示例 LitJson官方下载链接
- excel生成二维码
- Eclipse中如何clean项目以及clean的作用
- 手机内存不够用,蒲公英X1让U盘秒变私有云
- OkHttp3源码详解(四)缓存策略,万分膜拜
- 一 、单片机IO实验
- 使用OpenCV训练好的级联分类器识别人脸
- scrollView封装广告轮播实例代码(Hank版)
- 70-webApp打包:将网站利用hbuilder打包为安卓app
- RedHat6.2服务器配置方案大全--第一章:DNS
热门文章
- iMacCleaner for Mac(系统清理工具)
- 劳务网策划书(初稿)
- 国际期货资产管理|期货仓单|期货分仓软件|期货仓单资管软件
- 对抗自编码器AAE——pytorch代码解读试验
- ZED2双目相机kalibr标定终于得到了个相对满意的结果
- 51单片机扩展io口C语言编写,51单片机扩展IO口后扩展口的地址如何确定
- idhttpserver接收数据
- 又见大裁员,WeWork计划裁员比例可能达25%!|一周硅谷热闻
- 秀儿!用SSM框架实现了支付宝的支付功能,神操作啊
- php盯盘,操盘手教你盯盘:盘口挂单之分时图盯盘界面