博客專欄

        EEPW首頁 > 博客 > AnimeGANv2 實現動漫風格遷移,簡單操作

        AnimeGANv2 實現動漫風格遷移,簡單操作

        發布人:AI科技大本營 時間:2022-02-12 來源:工程師 發布文章

        作者 | Yunlord

        出品 | CSDN博客

        前言

        之前一直在研究如何將圖像動漫化,嘗試了阿里云api和百度api,效果都不盡如人意。結果發現了一個寶藏github項目——AnimeGANv2,能夠將現實世界場景照片進行動漫風格化。

        可以看出AnimeGAN的效果非常好,而在去年九月發布的 AnimeGANv2 優化了模型效果,解決了 AnimeGAN 初始版本中的一些問題。 

        相比AnimeGAN,改進方向主要在以下4點:

        解決了生成的圖像中的高頻偽影問題。

        它易于訓練,并能直接達到論文所述的效果。

        進一步減少生成器網絡的參數數量。(現在生成器大小 8.17Mb)

        盡可能多地使用來自BD電影的新的高質量的風格數。

        那接下來我們就介紹如何在本地使用該項目。

        GitHub鏈接如下:

        http://https//github.com/TachibanaYoshino/AnimeGANv2

        一、安裝

        本文介紹的是基于PyTorch的實現,項目地址為animegan2-pytorch。

        首先需要配置好適合該項目的環境,具體需要的安裝步驟如下:

        微信圖片_20220212154026.jpg

        首先,將該項目clone到本地或者下載下來:

        git clone https://github.com/bryandlee/animegan2-pytorch


        然后,進入到本地的項目中,會看到如下的項目結構:

        ├── convert_weights.py
        ├── model.py
        ├── README.md
        ├── samples
        │   ├── compare
        │   │   ├── 1.jpg
        │   │   ├── 2.jpg
        │   │   └── 3.jpg
        │   ├── face_results.jpg
        │   ├── faces
        │   └── inputs
        │       ├── 1.jpg
        │       ├── 2.jpg
        │       └── 3.jpg
        │   ├── results
        ├── weights
        │   ├── celeba_distill.pt
        │   ├── paprika.pt
        │   ├── face_paint_512_v1.pt
        │   └── face_paint_512_v2.pt
        ├── hubconf.py
        └── test.py

        該項目的核心工作就是使用PyTorch構建的生成器模型,并將原始TensorFlow的模型參數轉換為了PyTorch模型參數。其中構建模型的腳本為model.py,而轉換模型的腳本為convert_weights.py,然后test.py則是封裝的一個的動漫化功能接口。

        因為下載下來的項目中的weights文件夾下包含了已經轉換好的pytorch參數,所以我們不需要進行轉化。

        至此,本項目環境配置就完成了,接下來就是使用該項目。

        二、現實圖片動漫化

        只需要調用test.py就可以使用該項目,具體的命令格式如下。

        python test.py --checkpoint [模型文件路徑] --input_dir [輸入圖像所在目錄] --output_dir [輸出目錄] --device [設備選擇,cpu或者cuda]

        其中weights文件夾下包含四個權重,其中celeba_distill.pt和paprika.pt是風景動漫遷移的權重,而face_paint_512_v1.pt和face_paint_512_v2.pt是人臉動漫遷移的權重。只需要選擇對應的權重就能實現自己想要的動漫風格遷移。

        最終我們看下效果。

        三、實現效果

        人物:

        微信圖片_20220212154107.jpg微信圖片_20220212154111.jpg

        總結

        本文簡單介紹如何使用AnimeGANv2這個開源項目來實現現實圖片轉換為動漫風格圖像,該項目對應的模型很輕量,希望感興趣的小伙伴們也能夠用起來,生成自己喜歡的動漫圖像。

        *博客內容為網友個人發布,僅代表博主個人觀點,如有侵權請聯系工作人員刪除。



        關鍵詞: AI

        相關推薦

        技術專區

        關閉
        主站蜘蛛池模板: 兴隆县| 夏河县| 隆尧县| 英山县| 罗源县| 攀枝花市| 大同市| 康定县| 扎囊县| 健康| 阿拉善盟| 珠海市| 罗山县| 济源市| 三门县| 潜江市| 烟台市| 安化县| 高淳县| 恭城| 双柏县| 云梦县| 新竹县| 山丹县| 沭阳县| 吉木乃县| 竹山县| 长阳| 长海县| 襄樊市| 梁河县| 准格尔旗| 石渠县| 贵南县| 铁岭市| 太仓市| 攀枝花市| 沙雅县| 晋城| 赤壁市| 孟津县|