语义分割
这是一个训练语义隔离模型的存储库。 随着时间的流逝,它会得到改善。 ··
目录
关于该项目
这个项目开始时是我的硕士论文。 我将尽我所能继续改进它。
内置使用
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