TorchSeg:PyTorch中的快速模块化参考实现和语义分段算法的轻松培训 源码

火炬节 该项目旨在为使用PyTorch的语义细分模型提供快速的模块化参考实现。 强调 模块化设计:通过组合不同的组件轻松构建自定义的语义细分模型。 分布式培训:比多线程并行方法( )快60%以上,我们使用多处理并行方法。 多GPU训练和推理:支持不同的推理方式。 提供预训练的模型并实现不同的语义细分模型。 先决条件 PyTorch 1.0 pip3 install torch torchvision Easydict pip3 install easydict 忍者sudo apt-get install ninja-build tqdm pip3 install tqdm 更新