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

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

1.準(zhǔn)備

開始之前，你要確保Python和pip已經(jīng)成功安裝在電腦上，如果沒有，可以訪問(wèn)這篇文章：超詳細(xì)Python安裝指南 進(jìn)行安裝。

**(可選1) **如果你用Python的目的是數(shù)據(jù)分析，可以直接安裝Anaconda：Python數(shù)據(jù)分析與挖掘好幫手—Anaconda，它內(nèi)置了Python和pip.

**(可選2) **此外，推薦大家用VSCode編輯器，它有許多的優(yōu)點(diǎn)：Python 編程的最好搭檔—VSCode 詳細(xì)指南。

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

conda env create -f environment.yml

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

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

作者使用的是谷歌硬盤，如果你無(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ò) 反斜杠 "" 要換成 斜桿 "/" 。

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

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ò)程肯定沒這么簡(jiǎn)單，你會(huì)遇到不少問(wèn)題，但是我相信大部分都是存放圖片的目錄結(jié)構(gòu)上的問(wèn)題，大家如果有興趣可以動(dòng)手試試。