Python如何將你的照片轉(zhuǎn)化為“速寫(xiě)”

Photo-Sketching 一個(gè)能將照片的輪廓識(shí)別出來(lái)并將其轉(zhuǎn)化為“速寫(xiě)”型圖像的開(kāi)源模塊。

這個(gè)模塊的使用也相對(duì)簡(jiǎn)單，下面給大家?guī)先矫娴?a href="http://wenjunhu.com/v/" target="_blank">教程：

1.虛擬環(huán)境及依賴安裝

這個(gè)項(xiàng)目推薦大家直接用Anaconda進(jìn)行環(huán)境的構(gòu)建和開(kāi)發(fā)：Python數(shù)據(jù)分析與挖掘好幫手—Anaconda，因?yàn)樽髡咛峁┝艘粋€(gè) environment.yml 文件，你只需要輸入以下命令，就能一鍵安裝環(huán)境和依賴：

conda env create -f environment.yml

此外，推薦大家用VSCode編輯器來(lái)編寫(xiě)像這樣的小型Python項(xiàng)目：Python 編程的最好搭檔—VSCode 詳細(xì)指南。

2.下載預(yù)訓(xùn)練模型

作者已經(jīng)訓(xùn)練好了一些識(shí)別模型方便大家使用，可以在下列地址找到：
https://drive.google.com/file/d/1TQf-LyS8rRDDapdcTnEgWzYJllPgiXdj/view

作者使用的是谷歌硬盤(pán)，如果你無(wú)法科學(xué)上網(wǎng)，可以使用我提供的完整源代碼+預(yù)訓(xùn)練模型，在 **Python實(shí)用寶典 **公眾號(hào)后臺(tái)回復(fù)：**sketch **即可獲取。

下載完成后解壓文件，將 latest_net_D.pth 和 latest_net_G.pth 放置到 Checkpoints 文件夾下：

3.運(yùn)行預(yù)訓(xùn)練模型

接下來(lái)，我們需要修改使用預(yù)訓(xùn)練模型的啟動(dòng)腳本，這些腳本都放在 PhotoSketchscripts 下，我們需要使用的是 test_pretrained.cmd 或者 test_pretrained.sh 這兩個(gè)腳本。

如果你是 windows 系統(tǒng)，請(qǐng)修改 test_pretrained.cmd 腳本，重點(diǎn)是dataDir、results_dir、checkpoints_dir：

dataDir 指向到 PhotoSketch 所在的文件夾目錄，如果你是跟我一樣這么配的，results_dir 只需要配成 %dataDir%PhotoSketchResults 即可，checkpoints_dir 則為 %dataDir%PhotoSketchCheckpoints 。

如果你是macOS或者Linux，則修改 test_pretrained.sh 文件，修改方法與上面windows 的一樣，只不過(guò) 反斜杠 "" 要換成 斜桿 "/" 。

修改完腳本后，打開(kāi)命令行/終端，輸入以下命令，就會(huì)將你 PhotoSketchexamples 目錄下的文件轉(zhuǎn)化為“速寫(xiě)”。

windows:

scriptstest_pretrained.cmd

Linux/MacOS:

./scripts/test_pretrained.sh

轉(zhuǎn)化結(jié)果可以在 PhotoSketchResults 中看到，如下兩圖所示。

待轉(zhuǎn)化目錄：

轉(zhuǎn)化后：

可以看到效果其實(shí)不是非常好，由于是作者預(yù)訓(xùn)練的模型，所以效果不好也正常，如果大家需要的話，可以自己針對(duì)性地拿一些圖像訓(xùn)練模型，并針對(duì)性地做識(shí)別，這樣做效果才是最好的。

你需要訓(xùn)練或測(cè)試自己的模型也非常簡(jiǎn)單：

• 在倉(cāng)庫(kù)的 根目錄中 ，運(yùn)行 scripts/train.sh 可以訓(xùn)練模型
• 在倉(cāng)庫(kù)的 根目錄中 ，運(yùn)行 scripts/test.sh 可以測(cè)試val集或測(cè)試集

當(dāng)然訓(xùn)練過(guò)程肯定沒(méi)這么簡(jiǎn)單，你會(huì)遇到不少問(wèn)題，但是我相信大部分都是存放圖片的目錄結(jié)構(gòu)上的問(wèn)題，大家如果有興趣可以動(dòng)手試試。