Python實現(xiàn)照片卡通化

更新時間：2021年12月06日 09:44:29 作者：劍客阿良_ALiang

animegan2-pytorch機器學習項目可以實現(xiàn)照片動漫化，本文將為大家詳細介紹一下如何使用這一項目，感興趣的小伙伴快來跟隨小編一起學習吧

前言

之前看到一個有意思的開源項目，主要是可以將一張照片變成卡通漫畫的風格。下面給大家放幾張官方給出的部分效果圖。

看到這個效果圖，還是非常經(jīng)驗的。下面我會分享一下這個項目，并且選擇一些我自己找的圖片試驗一下。

項目Github地址：github地址?

環(huán)境部署

先使用git將項目下載下來?？匆幌马椖拷Y(jié)構(gòu)。

我們看一下需要的環(huán)境。

非常簡單，只要pytorch就可以了。如果有不太會安裝的，可以參考：機器學習基礎(chǔ)環(huán)境部署?

項目使用

看一下readme怎么說明的。

直接使用命令，或者代碼執(zhí)行都可以，我們先看看如果命令操作的話都有哪些參數(shù)支持。

if __name__ == '__main__':
 
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--checkpoint',
        type=str,
        default='./weights/paprika.pt',
    )
    parser.add_argument(
        '--input_dir', 
        type=str, 
        default='./samples/inputs',
    )
    parser.add_argument(
        '--output_dir', 
        type=str, 
        default='./samples/results',
    )
    parser.add_argument(
        '--device',
        type=str,
        default='cuda:0',
    )
    parser.add_argument(
        '--upsample_align',
        type=bool,
        default=False,
        help="Align corners in decoder upsampling layers"
    )
    parser.add_argument(
        '--x32',
        action="store_true",
        help="Resize images to multiple of 32"
    )
    args = parser.parse_args()
    
    test(args)

可以通過項目結(jié)構(gòu)看到，作者提供了4個模型文件可供選擇，可以用--checkpoint參數(shù)切換。需要提供圖片的輸入文件夾，修改--input_dir參數(shù)即可，經(jīng)過我自己的幾次驗證，發(fā)現(xiàn)有的模型適合風景類型的圖片，有的適合人像，下面我分成兩部使用一下。