昨日,语义分割算法DFN、BiSeNet 第一作者ycszen开源了TorchSeg项目,基于PyTorch的快速模块化语义分割开源库,复现了DFN, BiSeNet, PSPNet算法。
该项目主要包括以下highlights: 1.模块化设计:通过组合不同的组件,轻松构建定制的语义分割模型(这个很赞有木有~); 2.支持分布式训练:使用multi-processing并行方法,比多线程并行方法快60%; 3.支持多GPU训练和推断:支持不同的推断方式; 4.提供预训练的模型并实现不同的语义分割模型。
BenchmarkSS:Single Scale MSF:Multi-scale + Flip 项目地址: https://github.com/ycszen/TorchSeg
欢迎大家 Star 和 Contribute!
|