五月GitHub最热机器学习项目都在这里了
本文经AI新媒体量子位(公众号ID:QbitAI)授权转载,转载请联系出处。 您的五月余额已不足,这个月最新最火的机器学习项目都看过了吗? Mybridge AI博客从将近250个机器学习开源项目中找到了标星数排名最靠前的Top 10项目,涵盖视觉问答、对象检测、自动生成评论等多个维度。 一起来看看吧~ 第1名:PythiaPythia是Facebook人工智能实验室出品的视觉和语言多模式研究的模块化框架,2138星。 Pythia基于PyTorch,支持Model Zoo、多任务,有各种内置数据集比如VQA、VizWiz,支持基于DataParallel和DistributedDataParallel的分布式训练,还可以实现高度自定义。 另外,拿Pythia来做TextVQA和VQA的入门代码库也不错。 链接: https://github.com/facebookresearch/pythia 第2名:云注释(Cloud Annotations)自定义对象检测和分类训练,2014星。 基于IBM云对象存储,在TensorFlow上训练,需要先在线进行图像标注,macOS,Windows和Linux都可以用。 链接: https://github.com/cloud-annotations/training 第3名:PySOT商汤出品,可以在视频里追踪单个对象,实现SiamRPN和SiamMask等算法,1703星。 PySOT实现了最先进的单一对象跟踪算法,包括SiamRPN和SiamMask等,用Python编写,在PyTorch上运行。 链接: https://github.com/STVIR/pysot 第4名:PyTorch-BigGraph大规模图形结构数据生成嵌入软件,1417星,同样也是Facebook出品,主要作者是巴黎的Luca Wehrstedt。 PyTorch-BigGraph是一个分布式系统,可以搞定多达数十亿实体和数万亿边缘的大型网络交互图形。 用PyTorch-BigGraph的话需要Python 3.6或更高版本,最低PyTorch 1.0,不需要投喂GPU,但是比较费CPU。 链接: https://github.com/facebookresearch/PyTorch-BigGraph 第5名:InterpretML - Alpha微软出品,用来训练可解释的机器学习模型,1039星。 这里的“可解释”指的是可以解释模型调试中遇到的错误,模型有没有搞歧视,如何让人类理解这个模型,模型是否合法,在医疗、司法等高风险的领域靠不靠谱。 链接: https://github.com/microsoft/interpret 第6名:Weights & Biases可视化和追踪机器学习实验工具,1098星。 分析机器学习实验,比TensorBoard更轻量级。每次运行时,可以保存超参数和输出指标,训练过程中能实现可视化模型,还能自动跟踪代码状态,系统指标和配置参数。 链接: https://github.com/wandb/client 第7名:MLIR“多级中间表示”编译器基础结构,832星。 TensorFlow的一部分。MLIR旨在成为一种混合IR(intermediate representation),可以支持统一基础架构中的多种不同要求,能够表示所有TensorFlow图,在TensorFlow图上实现优化和转换,用TensorFlow图或者TF Lite完成量化和图变换。 链接: https://github.com/tensorflow/mlir 第8名:MeshCNNPyTorch中三维网格的卷积神经网络,367星。 这是SIGGRAPH 2019上发表的一篇论文,可以用来做3D形状分割,作者来自以色列特拉维夫大学和亚马逊。 类似传统CNN,MeshCNN结合了在网格边缘上运行的专用卷积和池化层。卷积应用于边缘和它们的入射三角形的四个边缘,池化层用边缘折叠操作保留表面拓扑结构,给后面的层生成了网状连接。 链接: https://github.com/ranahanocka/MeshCNN 第9名:TensorWatch微软出品,用于深度学习和强化学习的调试,监控和可视化,299星。 TensorWatch是一款专为深度学习和强化学习而设计的debug和可视化工具。它充分利用Jupyter Notebook显示实时可视化,不需要一直翻日志就能实时查询训练过程,还可以用来构建自己的UI和仪表板。 此外,TensorWatch利用几个优秀的库来可视化模型图,审查模型统计,解释预测等。 链接: https://github.com/microsoft/tensorwatch 第10名:hncynic一个猎奇项目,自动生成hacker news风格的评论,完全无视文章本身,看一眼标题,剩下全靠编,240星。 来源靠的是hacker news评论数据,借此训练Transformer编码器-解码器模型,中途还顺手用了一下维基百科的数据。 生成的评论大多脑洞清奇。 比如这个新闻标题《用TensorFlow检测鱼》,生成的评论是酱婶的:
看起来好像很有道理的样子,然而完全不知道在说啥,而且是满满的杠精画风啊! 链接: https://github.com/leod/hncynic 传送门原文链接: https://medium.mybridge.co/machine-learning-open-source-for-the-past-month-v-may-2019-bf4ff9b80b1b
(编辑:ASP站长网) |