神經網絡寫詩(charRNN)


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 編寫的語音識別。完善了本項目(本項目已囊括圖像,文本,語音三大領域的例子)。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM