易学智能

 找回密码
 立即注册

QQ登录

只需一步,快速开始

查看: 2603|回复: 3

TorchSeg—基于PyTorch的快速模块化语义分割开源库

[复制链接]

81

主题

162

帖子

493

积分

推广大使

积分
493
发表于 2019-1-25 19:50:26 | 显示全部楼层 |阅读模式
昨日,语义分割算法DFN、BiSeNet 第一作者ycszen开源了TorchSeg项目,基于PyTorch的快速模块化语义分割开源库,复现了DFN, BiSeNet, PSPNet算法。

11.jpg

该项目主要包括以下highlights:
1.模块化设计:通过组合不同的组件,轻松构建定制的语义分割模型(这个很赞有木有~);
2.支持分布式训练:使用multi-processing并行方法,比多线程并行方法快60%;
3.支持多GPU训练和推断:支持不同的推断方式;
4.提供预训练的模型并实现不同的语义分割模型。

Benchmark
SS:Single Scale MSF:Multi-scale + Flip
22.jpg
33.jpg
44.jpg
55.jpg
项目地址:
https://github.com/ycszen/TorchSeg

欢迎大家 Star 和 Contribute!


回复

使用道具 举报

665

主题

1234

帖子

6683

积分

xdtech

Rank: 5Rank: 5

积分
6683
发表于 2019-2-20 23:32:09 | 显示全部楼层
这个很强啊
回复

使用道具 举报

665

主题

1234

帖子

6683

积分

xdtech

Rank: 5Rank: 5

积分
6683
发表于 2019-2-20 23:32:22 | 显示全部楼层
语义分割
现在的热点
回复

使用道具 举报

665

主题

1234

帖子

6683

积分

xdtech

Rank: 5Rank: 5

积分
6683
发表于 2019-2-20 23:32:26 | 显示全部楼层
也是越来越难了
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|易学智能

GMT+8, 2024-12-22 11:57 , Processed in 0.021519 second(s), 22 queries .

Powered by Discuz! X3.4

© 2001-2017 Comsenz Inc.

快速回复 返回顶部 返回列表