Python+AI實(shí)現(xiàn)給老照片上色
哈嘍，大家好。
今天繼續(xù)給大家分享有趣的 AI 項(xiàng)目。
上次我們分享了用GAN（生成對(duì)抗網(wǎng)絡(luò)）讓靜態(tài)圖片動(dòng)起來(lái)。
今天我們分享用NoGAN的圖像增強(qiáng)技術(shù)給老照片著色。效果如下：
原圖
上色后
NoGAN是一種新型GAN，它能花費(fèi)最少的時(shí)間進(jìn)行GAN訓(xùn)練。
今天分享的這個(gè)項(xiàng)目已在GitHub開(kāi)源項(xiàng)目，下面我們來(lái)運(yùn)行一下。
1. 準(zhǔn)備工作
首先，用git clone命令下載源碼
git?clone?https://github.com/jantic/DeOldify.git
進(jìn)入項(xiàng)目根目錄，安裝Python依賴包
pip3?install?-r?requirements.txt
編寫(xiě)代碼運(yùn)行項(xiàng)目之前，需要下載預(yù)訓(xùn)練好的模型。項(xiàng)目提供了三個(gè)模型
模型
區(qū)別如下：
- ColorizeArtistic_gen.pth：在有趣的細(xì)節(jié)和活力方面實(shí)現(xiàn)了最高質(zhì)量的圖像著色效果，該模型在 UNet 上使用 resnet34 為主干，通過(guò) NoGAN 進(jìn)行了 5 次評(píng)論家預(yù)訓(xùn)練/GAN 循環(huán)重復(fù)訓(xùn)練
- ColorizeStable_gen.pth：在風(fēng)景和肖像方面取得了最佳效果，該模型在 UNet 上使用 resnet101 為主干，通過(guò) NoGAN 進(jìn)行了 3 次評(píng)論家預(yù)訓(xùn)練/GAN 循環(huán)重復(fù)訓(xùn)練
- ColorizeVideo_gen.pth：針對(duì)流暢的視頻進(jìn)行了優(yōu)化，它僅使用初始生成器/評(píng)論家預(yù)訓(xùn)練/GAN NoGAN 訓(xùn)練。由于追求流暢的速度，它的色彩比前兩者少。
將下載好的模型文件放在項(xiàng)目根目錄的models目錄下即可。
2. 編寫(xiě)代碼
在項(xiàng)目根目錄同級(jí)目錄下創(chuàng)建Python文件，編寫(xiě)代碼加載剛剛下載好的模型文件。
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指定項(xiàng)目根目錄，weights_name指定接下來(lái)使用哪個(gè)模型為照片上色。
讀取老照片，進(jìn)行上色
import?cv2
import?numpy?as?np
from?PIL?import?Image
img?=?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模塊將圖片轉(zhuǎn)換成模型輸入所需要的格式，送入模型進(jìn)行上色，完成后保存。
上述代碼是我從項(xiàng)目源碼中抽取的，可以看到，運(yùn)行代碼還是非常簡(jiǎn)單的。
到此這篇關(guān)于Python+AI實(shí)現(xiàn)給老照片上色的文章就介紹到這了,更多相關(guān)Python AI老照片上色內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了Python數(shù)據(jù)類型字典dictionary，字典是另一種可變?nèi)萜髂Ｐ?，且可存?chǔ)任意類型對(duì)象。本篇文字將詳細(xì)講述字典的常用方法，需要的可以參考一下2022-03-03
Python3.6安裝卸載、執(zhí)行命令、執(zhí)行py文件的方法詳解
這篇文章主要介紹了Python3.6之安裝卸載、執(zhí)行命令、執(zhí)行py文件的方法，本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-02-02
Django的ORM模塊是框架特色功能之一，它把數(shù)據(jù)表與Python類對(duì)應(yīng)、表字段與類屬性對(duì)應(yīng)、類實(shí)例與數(shù)據(jù)記錄對(duì)應(yīng)，并將對(duì)類實(shí)例的操作映射到數(shù)據(jù)庫(kù)中，這篇文章主要介紹了Django?ORM基本應(yīng)用與原理剖析,需要的朋友可以參考下2022-10-10
這篇文章主要介紹了深入理解Python中的*重復(fù)運(yùn)算符，小編覺(jué)得還是挺不錯(cuò)的，這里分享給大家，供需要的朋友參考。2017-10-10
詳解Python 調(diào)用C# dll庫(kù)最簡(jiǎn)方法
這篇文章主要介紹了詳解Python 調(diào)用C# dll庫(kù)最簡(jiǎn)方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-06-06
關(guān)于python通過(guò)新建環(huán)境安裝tfx的問(wèn)題
這篇文章主要介紹了python安裝tfx/新建環(huán)境,新建一個(gè)環(huán)境tfx專門(mén)用來(lái)運(yùn)行流水線，這個(gè)環(huán)境安裝python3.8，對(duì)python安裝tfx相關(guān)知識(shí)感興趣的朋友一起看看吧2022-05-05
學(xué)習(xí)python 之編寫(xiě)簡(jiǎn)單乘法運(yùn)算題
這篇文章主要介紹了學(xué)習(xí)python 第一季 編寫(xiě)簡(jiǎn)單乘法運(yùn)算題,需要的朋友可以參考下2016-02-02
Restful_framework視圖組件代碼實(shí)例解析
這篇文章主要介紹了Restful_framework視圖組件代碼實(shí)例解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-11-11
解決pyshp UnicodeDecodeError的問(wèn)題
今天小編就為大家分享一篇解決pyshp UnicodeDecodeError的問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-12-12
python使用遞歸實(shí)現(xiàn)斐波那契數(shù)列的示例詳解
這篇文章主要給大家介紹了python使用遞歸實(shí)現(xiàn)斐波那契數(shù)列的示例,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者使用Python具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面來(lái)一起來(lái)學(xué)習(xí)吧2024-01-01