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

Posted 2021-05-02 我爱计算机视觉

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

---

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

该项目主要包括以下highlights：

1.模块化设计：通过组合不同的组件，轻松构建定制的语义分割模型（这个很赞有木有~）；

2.支持分布式训练：使用multi-processing并行方法，比多线程并行方法快60％；

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

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

Benchmark

SS:Single Scale MSF:Multi-scale + Flip

https://github.com/ycszen/TorchSeg

欢迎大家 Star 和 Contribute！

加入语义分割专业群

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

（请务必注明:语义分割）

（不会时时在线，如果没能及时通过验证还请见谅）

更多技术干货，详见：

---

长按关注我爱计算机视觉

给我一个好看吧！