上传者: 42138525
|
上传时间: 2022-03-11 15:46:14
|
文件大小: 1.89MB
|
文件类型: -
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