一大波PyTorch图像分割模型来袭,俄罗斯程序员出品新model zoo
鱼羊 发自 凹非寺
量子位 报道 | 公众号 QbitAI
一个新的图像分割model zoo来啦!
一大波基于PyTorch的图像分割模型整理好了就等你来用~
这个新集合由俄罗斯的程序员小哥Pavel Yakubovskiy一手打造,包含四种模型架构和30种预训练骨干模型(backbone),官方文档列举了四条主要特点:
高级API(两行代码构建神经网络)
4种模型架构:Unet,Linknet,FPN,PSPNet
每个架构有30种可用的编码器
所有编码器都具有预先训练的权重,可实现更快、更好的收敛
授人以鱼不如授人以渔,放出了model zoo还不算,小哥还贴心地提供了如何使用CamVid数据集进行训练的示例。
效果怎么样
CamVid数据集是计算机视觉领域常用的一个数据集,通常用来进行街景分割。
俄罗斯小哥用这个数据集训练了一个能识别出图片中汽车的新模型。无论这辆车是隐藏在暗夜之中还是只有黄豆大小,都逃不出图像分割模型的手掌心。
构建模型时他选择了传奇的Unet架构,并使用了se_resnext50_32x4d这个预训练骨干模型。
听上去好像很厉害的样子,但群众都相信眼见为实,不看看模型训练效果如何,心里难免有一丢丢怀疑。
嗯,看来不管是昏天黑地还是目标只有一丁点大,新模型的预测结果都挺不错。
再加大难度呢?
量子位表示眼已瞎,模型是真强。
Pavel Yakubovskiy小哥还在Colab上放了用Catalyst(PyTorch上的高级框架)训练模型的代码,感兴趣的话可以直接在浏览器上跑一下。
入园指南
model zoo已开源,免票入场,快来围观。
先来了解一下都有哪些预训练模型可以用:
以及它们的权重:
pip一下:
$ pip install segmentation-models-pytorch
$ pip install git+https://github.com/qubvel/segmentation_models.pytorch
python库安装完成,就可以轻松地创建模型啦:
1import segmentation_models_pytorch as smp2model = smp.Unet()
根据训练任务的不同,可以通过调整骨干模型来改变网络结构,并且使用预训练权重来进行初始化:
1model = smp.Unet('resnet34', encoder_weights='imagenet')
也可以改变模型的输出类型:
1model = smp.Unet('resnet34', classes=3, activation='softmax')
所有的模型都有预训练编码器,因此准备数据的方法必须与进行权重训练时一致:
1from segmentation_models_pytorch.encoders import get_preprocessing_fn2preprocess_input = get_preprocessing_fn(‘renset18’, pretrained=’imagenet’)
One More Thing
其实Pavel Yakubovskiy小哥早在去年就发布了一个基于Keras的图像分割model zoo,一经推出就颇受欢迎。
一样的配方,一样的味道,简单易用,一脉相承。麻麻再也不用担心我找不到合适的预训练模型。
不过基于Keras的版本预训练模型要少一些,小哥也没有放出对应的示例,效果如何,就需要自己去检验啦~
传送门
PyTorch版Github地址:
https://github.com/qubvel/segmentation_models.pytorch
Keras版Github地址:
https://github.com/qubvel/segmentation_models
Colab上手:
https://colab.research.google.com/gist/Scitator/e3fd90eec05162e16b476de832500576/cars-segmentation-camvid.ipynb#scrollTo=A40deTWg-WJ1
— 完 —
订阅AI内参,获取AI行业资讯
加入社群
量子位AI社群开始招募啦,量子位社群分:AI讨论群、AI+行业群、AI技术群;
欢迎对AI感兴趣的同学,在量子位公众号(QbitAI)对话界面回复关键字“微信群”,获取入群方式。(技术群与AI+行业群需经过审核,审核较严,敬请谅解)
诚挚招聘
量子位正在招募编辑/记者,工作地点在北京中关村。期待有才气、有热情的同学加入我们!相关细节,请在量子位公众号(QbitAI)对话界面,回复“招聘”两个字。
量子位 QbitAI · 头条号签约作者
վ'ᴗ' ի 追踪AI技术和产品新动态
喜欢就点「好看」吧 !
一大波PyTorch图像分割模型来袭,俄罗斯程序员出品新model zoo相关推荐
- 一大波优秀3D作品来袭!看各国3D艺术家如何描绘2020!
已经迈入农历2021年,回望过去一年,如果要用一张图像去表达你的2020会是怎么样的呢? 散落在世界各地的3D艺术家们用3D作品分享了他们心中的2020. 下面就让我们一起来欣赏这些优秀的3D创作吧! ...
- 勒索预警,近期一大波新型勒索病毒来袭
点击蓝字关注我们 目前勒索病毒仍然是全球最大的威胁,最近一年针对企业的勒索病毒攻击越来越多,大部分勒索病毒是无法解密的,一定要保持高度的重视,近期又有一大波新型勒索病毒来袭...... HildaCr ...
- 俄罗斯程序员编程水平全球第2名 哪国程序员编程水平全球第1名
点击上面 免费订阅本账号! 本公众号主要推送javaweb开发相关技术,基础知识点,同时会深入剖析复杂的问题,分享一些优秀的框架,大型项目经验,当今最流行的Javaweb技术,热点科技新闻,招聘信息, ...
- python新手程序员工资-程序员吐槽新同事:连我实习水平都不到,工资是我的1.7倍...
原标题:程序员吐槽新同事:连我实习水平都不到,工资是我的1.7倍 一位程序员网友郁闷地分享了自己最近和新同事一起工作的经历,并吐槽一心搞技术不如口才好,引来众网友热议. 这位网友的新同事是一位4年工作 ...
- DAPP(分布式应用),区块链新物种,程序员掘金新风口!
DAPP(分布式应用),区块链新物种,程序员掘金新风口! 何玺 百家号02-2211:24 区块链技术正在重构我们的世界.如果你不信,你回头看看过去10年互联网技术对这个世界的改变.今天何玺要聊的,是 ...
- 8 个适合程序员学习新技能的网站
点击上方"逆锋起笔",公众号回复 PDF 领取大佬们推荐的学习资料 程序员书库(ID:CodingBook) 猿妹编译 链接:https://medium.com/better-p ...
- PyTorch图像分割模型——segmentation_models_pytorch库的使用
一.概要 segmentation_models_pytorch是一个基于PyTorch的图像分割神经网络 这个新集合由俄罗斯的程序员小哥Pavel Yakubovskiy一手打造,对于图像分割而言简 ...
- 端午节一大波赠书福利惊喜来袭!
端午节+年中节 ,一大波福利来!不仅仅赠书,还要一大波优惠福利!赶紧来参与吧! 福利 1.留言下面列出书籍中你想要的图书,即有机会活动异步新书. ①.欢迎大家积极留言,从留言中选取点赞最多的两位,送 ...
- 一大波 Android 刘海屏来袭,全网最全适配技巧!
一.序 Hi,大家好,我是承香墨影! Apple 一直在引领设计的潮流,自从 iPhone X 发布之后,"刘海屏" 就一直存在争议.不过不管你怎样,Android 也要跻入 &q ...
最新文章
- Linux下sh/bash/source/.命令的区别(转)
- python 递归函数_连载|想用Python做自动化测试?递归函数
- Java----前端验证之验证码额实现
- java integer reverse_Leetcode7 Reverse Integer Java实现及分析
- jparepository查询所有_jparepository 怎样自定义查询
- matlab绘制不同线性的直方图,在matlab中绘制多个垂直直方图
- java集合框架之LinkedList
- Happy New Year!PR升3啦!
- 【日常小问题3】win10电脑忘记开机密码的解锁方法【转载】
- 跨站请求伪造(CSRF)攻击
- 已知ip地址和其子网掩码如何求网络号子网号主机号
- html5限制拖拽区域怎么实现,html5怎么实现拖拽
- 天池案例-贷款违约预测(lgb)
- P1419 寻找段落
- 信息无障碍研究机构---教育、公益等
- C语言笔试题(嵌入式软件开发)
- SAE英文会议论文模板及翻译
- PB 饼状图制作过程
- Android开发之仿QQ表情实现(上)
- 云服务器-裸金属介绍
热门文章
- 观看直播 | IoT in Action 微软深圳物联网大会开幕在即,精彩不打烊
- 区块链,到底是美好未来还是题材炒作?
- 滴滴怒怼美团;阿里麻吉宝刷屏;B站、爱奇艺上市 | CSDN 极客头条
- mysql负变量_MySQL的变量
- asp网站服务器软件,asp服务器软件
- linux ati显卡驱动下载,LINUX 下ati最新显卡驱动下载
- 街机三国服务器维护,街机三国4月2日07:00更新维护公告
- linux内存脚本下载,linux tmpfs及消耗内存脚本
- 三元一次方程组步骤_人教版数学七年级下册8.4三元一次方程组的解法精讲
- 一体机服务器性能,一体机的服务器