本文转自雷锋网,如需转载请至雷锋网官网申请授权。
请注意,下面的图示由Gregory Piatetsky绘制,每个库都有其类别,按星标和贡献者对其进行绘制,符号大小则以该库在Github上的提交次数的对数表示。
图1:深度学习,自然语言处理和计算机视觉的顶级Python库
按星级和贡献者数量绘制;按提交数的对数表示相对大小
那么事不宜迟 ,下面就是由KDnuggets员工精心挑选的30个可用于深度学习,自然语言处理和计算机视觉顶级Python库。
深度学习
1. TensorFlow
星标:149000,提交数:97741,贡献者:754
TensorFlow是针对机器学习的端对端开源平台。它具备综合灵活的工具、库和社区资源,可以帮助研究者推动先进的机器学习技术的发展以及开发者更轻松地开发和发布由机器学习支持的应用。
2. Keras
星标:50000,提交数:5349,贡献者:864
Keras是由python编写的机器学习API,其运行在机器学习的顶级平台TensorFlow上。
3. PyTorch
星标:43200,提交数:30696,贡献者:1619
使用Python实现的张量和动态神经网络,并带有强大的GPU加速。
4. fastai
星标:19800,提交数:1450,贡献者:607
通过利用当下最佳的技术实践,fastai极快地简化了训练过程也加速了神经网络。
5. PyTorch Lightning
星标:9600,提交数:3594,贡献者:317
<ahelp_1" name="help_1"><divfinished-text"><divch-text"></divch-text"></divfinished-text"></ahelp_1">
针对高性能AI研究的封装轻量版PyTorch。可缩小你的模型,而不是提供小的样板。
6. JAX
星标:10000,提交数:5708,贡献者:221
Python+NumPy程序的组合变换:区分、向量化、在GPU/TPU上的JIT等等。
7. MXNet
星标:19100,提交数:11387,贡献者:839
带有动态和突变感知的数据流管理调度程序的轻量、便捷、灵活的分布式/移动机器学习库:支持Python、R、Julia、Scala、Go、JavaScript等等。
8. Ignite
星标:3100,提交数:747,贡献者:112
可灵活透明地训练评估PyTorch神经网络的高级别库。
自然语言处理
9. FastText
星标:21700,提交数:379,贡献者:47
fastText是一个可高效学习单词表意和句子分类的库。
10. spaCy
星标:17400,提交数:11628,贡献者:482
使用Python和Cython实现的工业级自然语言处理库(NLP)
11. gensim
星标:11200,提交数:4024,贡献者:361
gensim通过使用大型语料库进行主题建模、文档索引和相似性检索。目标受众是自然语言处理(NLP)和信息检索(IR)社区
12. NLTK
星标:9300,提交数:13990,贡献者:319
NLTK——自然语言工具箱——是一整套针对自然语言处理研究和开发的开源Python模块、数据集和教程
13. Datasets(Huggingface)
星标:4300,提交数:568,贡献者:64
用于自然语言处理的使用PyTorch,TensorFlow,NumPy和Pandas工具的快速,高效,开放式数据集和评估指标
14. Tokenizers(Huggingface)
星标:3800,提交数:1252,贡献者:30
针对研究和生产的快速顶级的分词器
15. Transformers(Huggingface)
星标:3500,提交数:5480,贡献者:585
Transformers:针对Pytorch和TensorFlow 2.0 的顶级自然语言处理库
16. Stanza
星标:4800,提交数:1514,贡献者:19
针对很多人类语言的正式标准NLP Python库
17. TextBlob
星标:7300,提交数:542,贡献者:24
简单,Python风格,文本处理-情感分析,词性标记,名词短语提取,翻译等等。
18. PyTorch-NLP
星标:1800,提交数:442,贡献者:15
用于PyTorch自然语言处理(NLP)的基础实用工具
19. Textacy
星标:1500,提交数:1324,贡献者:23
用于执行各种自然语言处理(NLP)任务的Python库,构建在一个高性能spaCy库之上。
20. Finetune
星标:626,提交数:1405,贡献者:13
Finetune是一个库,它允许用户利用最新的预训练NLP模型来执行各种下游任务。
21. TextHero
星标:1900,提交数:266,贡献者:17
文本预处理、表示和可视化,从零到精通。
22. Spark NLP
星标:1700,提交数:4363,贡献者:50
Spark NLP是建立在Apache Spark ML之上的自然语言处理库。
23. GluonNLP
星标:2200,提交数:712,贡献者:72
GluonNLP这个工具包,可简化文本预处理,数据集加载和神经模型构建,以帮助您加快对自然语言处理(NLP)的研究。
计算机视觉
24. Pillow
星标:7800,提交数:10799,贡献者:303
Pillow是个对用户十分友好的PIL分支。PIL是Python图像库
25. OpenCV
星标:49600,提交数:29453,贡献者:1234
开源计算机视觉库
26. scikit-image
星标:4000,提交数:12352,贡献者:403
用Python做图像处理
27. Mahotas
星标:644,提交数:1273,贡献者:25
Mahotas是个包含了最快的计算机视觉算法(所有算法用C++实现保证运行速度)的库,针对numpy数组运行
28. Simple-CV
星标:2400,提交数:2625,贡献者:69
SimpleCV是一个针对开源机器视觉的架构,使用了OpenCV和Python编程语言。
29. GluonCV
星标:4300,提交数:774,贡献者:101
GluonCV提供了计算机视觉中最先进(SOTA)的深度学习模型。
30. Torchvision
星标:7500,提交数:1286,贡献者:334
Torchvision包包含了流行的数据集、模型架构和计算机视觉中常用的图像转换方法。