pytorch-SCNN
**于2018/12/15更新。
去做
支持不同的骨干网
支持城市景观
单GPU训练
介绍
这是的PyTorch(0.4.1)实现。它可以使用Modified Aligned ResNet作为主干。目前,我们使用Cityscapes数据集训练SCNN。
安装
该代码已通过Anaconda和Python 3.6进行了测试。安装Anaconda环境后:
克隆仓库:
git clone https://github.com/forlovess/SCNN-pytorch.git
cd SCNN-pytorch
安装依赖项:
有关PyTorch依赖关系,请参见以获得更多详细信息。
对于自定义依赖项:
pip install matplotlib pillow tensorboardX tqdm
训练
请按照以下步骤训练您的模型:
输入参数:(通过python tr
2022-03-11 15:46:14
1.89MB
Python
1