语义分割 这是一个训练语义隔离模型的存储库。 随着时间的流逝,它会得到改善。 ·· 目录 关于该项目 这个项目开始时是我的硕士论文。 我将尽我所能继续改进它。 内置使用 Python 3.7 PyTorch 1.2.0 火炬视觉0.4.0 入门 要启动并运行本地副本,请遵循以下简单步骤。 要求 首先按照以下步骤安装要求。 pip install -r requirments.txt 安装 克隆仓库git clone https://github.com/MR3z4/SemanticSegmentation.git 运行训练代码python main.py 用法 它将随着时间的推移完成。 路线图 多GPU支持 增加RMI损失 添加用于训练的混音选项 毫不犹豫地添加混音以进行混音训练 添加AdaBelief优化器选项进行培训 添加CE2P网络(具有正常的BatchNorm
1