https://github.com/chenyuntc/pytorch-book
基於pytorch ,許多有趣的小應用。感謝作者!
作者的代碼寫得非常清晰,配置方法也很明確,只需要按照提示,安裝依賴:
pip install -r requirements.txt
導入:from torchnet import meter 單獨下載:git+https://github.com/pytorch/tnt.git@master 解壓,取出 torchnet
然后,下載相應的雲訓練model,按照命令提示即可運行demo。
另外:Python3 運行時,將# coding:utf8 改為:# coding:utf-8
例如:
python main.py gen --model-path='checkpoints/tang_199.pth' --pickle-path='tang.npz' --start-words='我是少帥' --prefix-words='江流天地外,山色有無中。' --acrostic=True --nouse-gpu
結果:
我來冒山岳,舊隱來相暾。 是日不相識,白雲如一言。 少年不得意,一舉十二門。 帥來出中府,脫屐何時聞。
目前demo跑通第七章和第九章的model。
實戰部分(第六到十章)利用PyTorch實現了幾個酷炫有趣的應用,對於這部分的內容,本倉庫給出完整的實現代碼,並提供預訓練好的模型作為demo,供讀者測試。
- 第六章是承上啟下的一章,這一章的目標不是教會讀者新函數,新知識,而是結合Kaggle中一個經典的比賽,實現一個深度學習中比較簡單的圖像二分類問題。在實現過程中,帶領讀者復習前五章的知識,並提出代碼規范以合理的組織程序,代碼,使得程序更加可讀,可維護。第六章還介紹了在PyTorch中如何進行debug。
- 第七章為讀者講解了當前最火爆的生成對抗網絡(GAN),帶領讀者從頭實現一個動漫頭像生成器,能夠利用GAN生成風格多變的動漫頭像。
- 第八章為讀者講解了風格遷移的相關知識,並帶領讀者實現風格遷移網絡,將自己的照片變成高大上的名畫。
- 第九章為讀者講解了一些自然語言處理的基礎知識,並講解了CharRNN的原理。而后利用收集了幾萬首唐詩,訓練出了一個可以自動寫詩歌的小程序。這個小程序可以控制生成詩歌的格式,意境,還能生成藏頭詩。
- 第十章為讀者介紹了圖像描述任務,並以最新的AI Challenger比賽的數據為例,帶領讀者實現了一個可以進行簡單圖像描述的的小程序。
- 第十一章(新增,實驗性) 由Diamondfan 編寫的語音識別。完善了本項目(本項目已囊括圖像,文本,語音三大領域的例子)。