SCNN-pytorch:PyTorch中的空间CNN模型使用Cityscapes-dataset

时间:2024-04-07 03:28:34
【文件属性】:

文件名称:SCNN-pytorch:PyTorch中的空间CNN模型使用Cityscapes-dataset

文件大小:1.89MB

文件格式:ZIP

更新时间:2024-04-07 03:28:34

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)

网友评论