谷歌EfficientNet缩放模型,PyTorch实现出炉,登上GitHub热榜丨Demo可用
郭一璞 发自 凹非寺
量子位 报道 | 公众号 QbitAI
谷歌上个月底提出的EfficientNet开源缩放模型,在ImageNet的准确率达到了84.1%,超过Gpipe,已经是当前的state-of-the-art了。
出炉没几天,官方TensorFlow版本在GitHub上就有了1300+星。
现在,哈佛数学系小哥哥Luke Melas-Kyriazi开源了自己的PyTorch实现,包含与训练模型和Demo。
帖子一出,就收到了众多PyTorch用户的欢迎,在Reddit上拿到了超过170个点赞,不少用户都准备上手尝试了:
等不及想把它merge到torchvision里!
在GitHub上也登上了热榜。
甚至,这个PyTorch实现还在隔壁岛国的收到了热烈欢迎。
啥是EfficientNet
EfficientNets是一种新的模型缩放方法,准确率比之前最好的Gpipe提高了0.1%,但是模型更小更快,参数的数量和FLOPS都大大减少,效率提升了10倍。
用复合缩放(compound scaling)的方法,与缩放神经网络的传统方法不同,用一组固定的缩放系数统一缩放每个维度。
实现复合缩放的首先是执行网格搜索,以在固定资源约束下找到基线网络(baseline model)的不同缩放维度之间的关系,确定每个维度的缩放比例系数。然后将这些系数将应用于基线网络,扩展到所需的目标模型大小或计算力。
模型缩放的有效性也在很大程度上依赖于基线网络。因此,为了进一步提高性能,谷歌还使用AutoML MNAS框架优化了模型的准确率和效率,执行神经架构搜索来开发新的基线网络。
安装使用方式
可以使用pip安装
或者用源代码安装
1git clone https://github.com/lukemelas/EfficientNet-PyTorch2cd EfficientNet-Pytorch3pip install -e .
加载EfficientNet
1from efficientnet_pytorch import EfficientNet2model = EfficientNet.from_name(‘efficientnet-b0’)
加载预训练模型
1from efficientnet_pytorch import EfficientNet2model = EfficientNet.from_pretrained(‘efficientnet-b0’)
模型具体详情:
有Demo
Luke还准备了一份Colab笔记本Demo。
示例中,先悄咪咪的扔一只胖达。
嘿,果然认出来是胖达。
传送门
GitHub
https://github.com/lukemelas/EfficientNet-PyTorch
Google原论文
EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks
Mingxing Tan, Quoc V. Le
https://arxiv.org/abs/1905.11946
Colab Demo
https://colab.research.google.com/drive/1Jw28xZ1NJq4Cja4jLe6tJ6_F5lCzElb4
— 完 —
小程序|全类别AI学习教程
AI社群|与优秀的人交流
量子位 QbitAI · 头条号签约作者
վ'ᴗ' ի 追踪AI技术和产品新动态
喜欢就点「在看」吧 !
谷歌EfficientNet缩放模型,PyTorch实现出炉,登上GitHub热榜丨Demo可用相关推荐
- 用PyTorch实现的李沐《动手学深度学习》,登上GitHub热榜,获得700+星
晓查 发自 凹非寺 量子位 报道 | 公众号 QbitAI 李沐老师的<动手学深度学习>是一本入门深度学习的优秀教材,也是各大在线书店的计算机类畅销书. 作为MXNet的作者之一,李沐老 ...
- 李沐《动手学深度学习》PyTorch 实现版开源,瞬间登上 GitHub 热榜!
点击上方"AI有道",选择"星标"公众号 重磅干货,第一时间送达 李沐,亚马逊 AI 主任科学家,名声在外!半年前,由李沐.Aston Zhang 等人合力打造 ...
- PyTorch实现的李沐《动手学深度学习》,登上GitHub热榜,获得1000+星
点击我爱计算机视觉标星,更快获取CVML新技术 晓查 发自 凹非寺 量子位 报道 | 公众号 QbitAI 李沐老师的<动手学深度学习>是一本入门深度学习的优秀教材,也是各大在线书店的计 ...
- 最佳开发工具大全!前谷歌工程师两年打造“厂外生存指南”,登上GitHub热榜
晓查 栗子 方驭洋 发自 凹非寺 量子位 报道 | 公众号 QbitAI 一位曾经的谷歌工程师,花费两年时间,辛苦整理了一份清单. 这个名为"xg2xg"的清单,原本是这位前谷歌 ...
- 把命令行玩成“迷你谷歌”:可搜索、计算,还能翻译 | GitHub热榜
点击上方"AI遇见机器学习",选择"星标"公众号 重磅干货,第一时间送达 金磊 发自 凹非寺 量子位 报道 | 公众号 QbitAI 还在用浏览器打开谷歌.百度 ...
- 男神青涩时纤毫毕现!腾讯AI模型GFPGAN火上GitHub热榜第一,Demo在线可玩
包浆老图立刻清晰到头发丝是一种怎样的体验? 看这明亮的眼眸.清晰的发丝,原本模糊的人像立刻添了几分灵动: (这就是用AI修复的高圆圆童年照) 时间更久远的历史名人照片也能搞定. 鲁迅先生年轻时意气风发 ...
- 女神青涩时纤毫毕现!腾讯AI模型GFPGAN火上GitHub热榜第一,Demo在线可玩
明敏 发自 凹非寺 量子位 报道 | 公众号 QbitAI 包浆老图立刻清晰到头发丝是一种怎样的体验? 看这明亮的眼眸.清晰的发丝,原本模糊的人像立刻添了几分灵动: (这就是用AI修复的高圆圆童年照) ...
- 女神青涩时纤毫毕现,腾讯 AI 模型 GFPGAN 火上 GitHub 热榜第一,Demo 在线可玩
来源:量子位 作者:明敏 包浆老图立刻清晰到头发丝是一种怎样的体验?看这明亮的眼眸.清晰的发丝,原本模糊的人像立刻添了几分灵动(这就是用 AI 修复的高圆圆童年照). 甚至时间更久远的历史名人照片也能 ...
- 大小仅1MB,超轻量级通用人脸检测模型登上GitHub趋势榜
机器之心报道 项目作者:Linzaer 近日,用户 Linzaer 在 Github 上推出了一款适用于边缘计算设备.移动端设备以及 PC 的超轻量级通用人脸检测模型,该模型文件大小仅 1MB,320 ...
最新文章
- C语言学生学籍管理系统 链表+存储数据到文档
- 台虚拟机更改计算机名,检查虚拟机的硬件信息、更改虚拟机名称、查看虚拟机配...
- windows系统如何cmake_Windows操作系统如何快速增加分区?建议收藏
- 【集训队作业2018】复读机【指数型生成函数】【单位根反演】【二项式定理】
- 数据库笔记01:SQL Server系统概述
- webservice studio 参数是DataSet时不支持中文 解决方法
- maven学习笔记第一节一-maven install 模块之间相互引用
- 《高质量C/C++编程指南》学习笔记
- windows下客户端连接上马上会断开连接_zookeeper系列客户端的骚操作amp; Curator使用...
- (转)Fintech路上券商究竟做错了什么?漏做了什么?
- mysql外键必须连接主键_MySQL数据库的主键和外键详解3
- matlab做信号实验需要安装那些模块_Amesim软件安装过程
- gamepad android,传说之下虚拟手柄
- android 改机型玩王者,全机型60帧一键修改
- 【PyQt5】串口数据实时绘图
- pyecharts中文手册
- 剖析《口袋妖怪复刻版》技术架构
- ASR - OpenAI whisper
- [编程题]雀魂启动!
- jabref java_Jabref安装及使用教程
热门文章
- 背靠 Google 的 Go 语言,就不会失败?
- Java多线程同步数据库,源码+原理+手写框架
- android alsa,android下alsa驱动alsa_arecord录音问题
- c语言 字体点阵取模,[汉字取模软件怎么用]单片机点阵汉字显示需要用
- koa ajax,nodejs之koa2请求示例(GET,POST)
- java tcp demo_详解Java使用TCP实现数据传输实例
- mysql x key 组合_技本功丨浅谈MySQL的七种锁
- python怎么后撤步_你的后撤步用对了吗? | 后撤步的目的
- 怎么让同网络计算机强制关机,知道局域网ip怎么关机
- 李少华 linux内核,腾讯云CMQ消息队列在Linux环境下的使用