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

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

animegan2-pytorch機(jī)器學(xué)習(xí)項(xiàng)目可以實(shí)現(xiàn)照片動(dòng)漫化，本文將為大家詳細(xì)介紹一下如何使用這一項(xiàng)目，感興趣的小伙伴快來(lái)跟隨小編一起學(xué)習(xí)吧

前言

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

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

項(xiàng)目Github地址：github地址?

環(huán)境部署

先使用git將項(xiàng)目下載下來(lái)。看一下項(xiàng)目結(jié)構(gòu)。

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

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

項(xiàng)目使用

看一下readme怎么說(shuō)明的。

直接使用命令，或者代碼執(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)

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