優秀開源AI框架推薦

​

​

AI領域開源框架推薦

​

• AutoML開發框架

AutoML全稱是Automated Machine Learning，即自動機器學習，聽起來是不是很酷，沒錯的確很酷，如果感興趣的話可以讀一下我們實驗室寫的AutoML綜述： AutoML: A Survey of State-of-the-art。

https://zhuanlan.zhihu.com/p/158162306​zhuanlan.zhihu.com

​

至於開發框架主要以基於深度學習的為主推薦一下幾個：

1. 利益相關首先推薦 Vega，之前寫了一篇介紹Vega的文章，如下。簡而言之，Vega是目前第一個實現端到端AutoML的框架，目前正在參與實習工作，新的版本放出來后易用性相比於已放出的1.0版本會有很大提升，敬請期待。

https://zhuanlan.zhihu.com/p/258362976​zhuanlan.zhihu.com

2. 第二個推薦巨硬的 NNI, 易用性很強，下至胎兒，上至100歲都很容易上手。反正Vega和NNI選哪個都不差

microsoft/nni​github.com

3. 其他沒怎么用過，僅列出名字，歡迎大佬們評論區留言評價以下這些框架：

https://autokeras.com/​autokeras.com

AutoKeras​autokeras.com

awslabs/autogluon​github.com

​

• 3D Deep Learning

下面這個是最近由MIT大學韓松團隊開發的用於3D 深度學習場景的開源框架。

mit-han-lab/e3d​github.com

​

• 基於Pytorch深度定制的開發框架
• 首推Pytorch-lightning ，目前已經更新到1.0版本，這個庫的易用性是保姆級別的，TPU、GPU、多機多卡、amp等一大堆功能都給你弄好了

https://github.com/PyTorchLightning/pytorch-lightning​github.com

• torchline: 這個庫是我基於pytorch-lightning進一步封裝的，主要是引入了注冊機制，所有的模型、優化器、損失函數、數據集等都可以通過一個yml文件完成配置，不用再寫一大堆的argparse了，易用性進一步提高。

https://zhuanlan.zhihu.com/p/97244535​zhuanlan.zhihu.com

​

• fast.ai： 這個庫怎么說呢，我個人是不太喜歡用的，文檔寫的不合我的胃口。原因可以看這個回答，總之這個庫的tutorial里居然好多地方都是 from ... import * ，然后代碼示例里就一堆不知道從哪里來的函數和類，emm，我還得去源代碼里找具體是干啥的

如何評價Fastai？​www.zhihu.com

​

目標檢測

• 首推鼎鼎大名的mmdetection

  https://github.com/open-mmlab/mmdetection

• 當然Pytorch官方的 Detectron2 也是一個非常不錯的選擇

  https://github.com/facebookresearch/detectron2

• 聯邦學習開發框架: FedML.ai

https://zhuanlan.zhihu.com/p/262497372​zhuanlan.zhihu.com

• 圖形學開發框架：

taichi-dev/taichi​github.com

​

TaiChi是由胡淵鳴大佬基於Python開發，可以非常方便實現下面這些酷到不行的特效

而且Taichi作者也專門開設了 GAMES201課程

taichi-dev/games201​github.com

另外 GAMES101 系列課程也強烈安利，閆令琪老師講的炒雞贊，我也開設了專欄記錄課程筆記，歡迎品嘗

GAMES計算機圖形學入門學習​

​