前言

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

看到這個效果圖，還是非常經驗的，下面我會分享一下這個專案，并且選擇一些我自己找的圖片試驗一下，

專案Github地址：github地址

環境部署

先使用git將專案下載下來，看一下專案結構，

我們看一下需要的環境，

非常簡單，只要pytorch就可以了，如果有不太會安裝的，可以參考我的另一篇文章：機器學習基礎環境部署 | 機器學習系列_阿良的博客-CSDN博客

專案使用

看一下readme怎么說明的，

直接使用命令，或者代碼執行都可以，我們先看看如果命令操作的話都有哪些引數支持，

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)

可以通過專案結構看到，作者提供了4個模型檔案可供選擇，可以用--checkpoint引數切換，需要提供圖片的輸入檔案夾，修改--input_dir引數即可，經過我自己的幾次驗證，發現有的模型適合風景型別的圖片，有的適合人像，下面我分成兩部使用一下，