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

点击我爱计算机视觉标星,更快获取CVML新技术


昨日,语义分割算法DFN、BiSeNet 第一作者ycszen开源了TorchSeg项目,基于PyTorch的快速模块化语义分割开源库,复现了DFN, BiSeNet, PSPNet算法。

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

该项目主要包括以下highlights:

1.模块化设计:通过组合不同的组件,轻松构建定制的语义分割模型(这个很赞有木有~);

2.支持分布式训练:使用multi-processing并行方法,比多线程并行方法快60%;

3.支持多GPU训练和推断:支持不同的推断方式;

4.提供预训练的模型并实现不同的语义分割模型。

Benchmark

SS:Single Scale MSF:Multi-scale + Flip

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

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

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

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

项目地址:

https://github.com/ycszen/TorchSeg

欢迎大家 Star 和 Contribute!

加入语义分割专业群

加入52CV-语义分割专业讨论群,扫码添加52CV君(your-word)拉你入群:

(请务必注明:语义分割)

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

喜欢在QQ交流的童鞋,可以加52CV官方QQ群:928997753。

(不会时时在线,如果没能及时通过验证还请见谅)

更多技术干货,详见:

"我爱计算机视觉"干货集锦分类汇总(2019年1月20日)


长按关注我爱计算机视觉

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

给我一个好看吧