semseg:pytorch中的语义分割

时间:2024-05-22 03:46:49
【文件属性】:

文件名称:semseg:pytorch中的语义分割

文件大小:1.4MB

文件格式:ZIP

更新时间:2024-05-22 03:46:49

Python

PyTorch语义分割 介绍 该存储库是用于语义分割/场景解析的PyTorch实现。 该代码易于在各种数据集上进行训练和测试。 该代码库主要使用ResNet50 / 101/152作为主干,并且可以轻松地适应其他基本分类结构。 已实施的网络包括和 ,它们在 , 和。 样本实验数据集为 , 和 。 更新 2020.05.15:分支master ,使用官方的 ,仅支持多处理培训,并使用pytorch 1.4.0进行了测试。 2019.05.29:分支1.0.0支持多线程训练( )和多处理训练( )(推荐)。 而后一个要快得多。 使用syncbn从和顶点,与pytorch 1.0.0测试。 用法 强调: 使用官方的nn.SyncBatchNorm快速进行多处理培训( nn.parallel.DistributedDataParallel )。 设计良好的代码结构可带来更好的重新实现结


【文件预览】:
semseg-master
----FAQ.md(7KB)
----model()
--------psanet.py(9KB)
--------pspnet.py(4KB)
--------resnet.py(7KB)
----lib()
--------psa()
----util()
--------util.py(6KB)
--------transform.py(10KB)
--------config.py(5KB)
--------dataset.py(3KB)
----config()
--------voc2012()
--------cityscapes()
--------ade20k()
----LICENSE(1KB)
----README.md(9KB)
----figure()
--------pspnet.png(1.33MB)
--------demo()
----data()
--------camvid()
--------voc2012()
--------cityscapes()
--------bdd()
--------ade20k()
----tool()
--------train.py(20KB)
--------test.py(11KB)
--------train.sh(725B)
--------demo.py(8KB)
--------test.sh(559B)
----.gitignore(815B)

网友评论