超贊的PyTorch資源大列表,GitHub標星9k+
自 2017 年 1 月 PyTorch 推出以來,其熱度持續(xù)上升。PyTorch 能在短時間內(nèi)被眾多研究人員和工程師接受并推崇是因為其有著諸多優(yōu)點,如采用 Python 語言、動態(tài)圖機制、網(wǎng)絡構建靈活以及擁有強大的社群等。
最近紅色石頭在瀏覽 GitHub 的時候發(fā)現(xiàn)了一份超贊的 Pytorch 資源大列表,基本上把 PyTorch 實現(xiàn)的 NLP/CV 相關項目、示例代碼、相關庫、論文實現(xiàn)等都包含了。這里我做了整理,推薦給大家。
巧的是,看到這份資源大列表比較熟悉,對比之前的文章發(fā)現(xiàn),原來之前 GitHub 上有原始的英文版,這份是漢化翻譯的,良心啊~
原來的英文版 GitHub 項目地址:
https://github.com/bharathgs/Awesome-pytorch-list
本文推薦的漢化版 GitHub 地址:
https://github.com/xavier-zy/Awesome-pytorch-list-CNVersion
目錄
這份資源大列表主要包含下面的內(nèi)容:
1. PyTorch 與相關庫
- 自然語言處理與語音處理
- 計算機視覺
- 概率庫與生成庫
- 其他庫
2. 教程與示例3. 論文實現(xiàn)4. 報告與會議5. 其它
下面分別介紹!
1. 自然語言處理與語音處理
這部分包含了 41 個 PyTorch NLP 相關的熱門項目,例如一種跨多說話者的語音生成方法、語音轉文字的端到端模型實現(xiàn)、快速生成WaveNet的實現(xiàn);PyTorch NLP 相關的熱門庫,例如基于 FastAI 的 Pytorch NLP 庫、用來計算和使用多語言語句嵌入的庫 LASER;PyTorch NLP 相關的熱門框架和工具,例如pytorch-seq2seq、PyTorch 中的 Sequence-to-Sequence 框架 nmtpytorch等等。
值得一提的是,這些項目很多都是官方的實現(xiàn),一般會有系統(tǒng)的使用說明,包含了安裝、加載、訓練、測試和演示等多方面的詳細解釋。而且官方也一直在更新,非常不錯。
2. 計算機視覺
這部分包含了 25 個 PyTorch CV 相關的熱門項目和庫。例如包含流行的數(shù)據(jù)集、模型架構、計算機視覺中常用的圖像變換的 TorchVision,用于機器學習的圖像增強庫 Augmentor,實例分割與對象檢測的快速模塊化參考實現(xiàn) maskrcnn-benchmark,基于 PyTorch 的 2D 和 3D 面部對齊庫 ace-alignment 等等。
該部分項目主要涉及神經(jīng)風格遷移、圖像分類、人臉對齊、語義分割、RoI 計算、圖像增強等任務,還有一些特殊的 CNN 架構。
3. 概率庫和生成庫
這部分主要包括概率編程和統(tǒng)計推斷、生成概率庫、PyTorch 中的貝葉斯優(yōu)化等。
4. 教程與示例
這部分主要介紹 66 個 PyTorch 經(jīng)典教程,包括強化學習、NLP、CV。Logistic、CNN、RNN、LSTM 等神經(jīng)網(wǎng)絡模型由數(shù)行代碼實現(xiàn),一些高級示例由復雜模型實現(xiàn)。
這份教程示例清單基本涵蓋了 PyTorch 各種教程,難易程度不同,適合剛入門的新手,也有適合進階學習的。
例如,第 5 個是 PyTorch 各種教程,在其官方教程里,內(nèi)容豐富:
https://pytorch.org/tutorials/
可謂最經(jīng)典的 PyTorch 入門教程《Deep Learning with PyTorch: A 60 Minute Blitz》就出自這里。
5. 論文實現(xiàn)
這部分包括了 338 個 PyTorch 相關的論文實現(xiàn)。例如用PyTorch實現(xiàn)的產(chǎn)生序列數(shù)據(jù)的遞歸變分自動編碼器、PyTorch 實現(xiàn) V-Net:全卷積神經(jīng)網(wǎng)絡在體醫(yī)學圖像分割中的應用、生成對抗網(wǎng)絡的 PyTorch 簡單實現(xiàn),關注于動漫臉譜繪畫等等。
6. 其它
這部分主要介紹了 37 個 PyTorch 各種各種資源,包括教程、論文、項目、社區(qū)等的清單,PyTorch 論壇、PyTorch 深度學習模板等。還有一些非常有意思的項目,例如使用神經(jīng)網(wǎng)絡作畫、用 PyTorch 實現(xiàn)的聊天機器人、用 AlphaZero 算法玩五子棋等。
總的來說,這是一份超贊的 PyTorch 資源大列表,非常全面。而且對原項目進行了翻譯。值得推薦!
最后再附上該項目的 GitHub 地址:
https://github.com/xavier-zy/Awesome-pytorch-list-CNVersion