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