【創新實訓】風格遷移功能探索與實現(六)完成代碼並訓練 保存訓練結果 測試效果

全部完成時代碼結構如下:  使用MS COCO數據集: 放到train2014下。 該數據集解壓後18G左右,共328000張圖片,每張style-image訓練的話,在服務器GPU環境下需要4-5個小時左右,迭代次數50000多次,最終Loss收斂在六七十。 測試使用模型: 可以看到,使用cubist風格轉換後還不錯 最終訓練結果:
相關文章
相關標籤/搜索