【文件属性】:
文件名称:SCNN-pytorch:PyTorch中的空间CNN模型使用Cityscapes-dataset
文件大小:1.89MB
文件格式:ZIP
更新时间:2021-03-18 09:41:54
Python
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
【文件预览】:
SCNN-pytorch-master
----.gitignore(1KB)
----modeling()
--------backbone()
--------__init__.py(0B)
--------SCNN.py(2KB)
--------build_rnn.py(6KB)
----LICENSE(1KB)
----doc()
--------results.png(1.66MB)
--------model.PNG(209KB)
----utils()
--------summaries.py(1KB)
--------saver.py(2KB)
--------calculate_weights.py(961B)
--------metrics.py(2KB)
--------lr_scheduler.py(3KB)
--------loss.py(2KB)
----README.md(2KB)
----_gitignore(1KB)
----dataloaders()
--------datasets()
--------__init__.py(831B)
--------custom_transforms.py(5KB)
--------utils.py(3KB)
--------combine_dbs.py(3KB)
----train.py(11KB)