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

昨日，语义分割算法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！