文件名称:semantic-segmentation:Nvidia语义分割monorepo
文件大小:2.55MB
文件格式:ZIP
更新时间:2024-05-21 13:40:02
Python
| |
本文Pytorch实现。
如果您要查找与相对应的代码,请参考sdcnet分支。
安装
该代码已使用pytorch 1.3和python 3.6进行了测试
您可以使用./Dockerfile生成映像。
下载权重
创建一个目录,您可以在其中保留大文件。 理想情况下,不在此目录中。
> mkdir < large>
更新config.py __C.ASSETS_PATH以指向该目录
__C.ASSETS_PATH = <大资产目录>
从下载预训练的权重,并放入
【文件预览】:
semantic-segmentation-main
----Dockerfile(827B)
----PREPARE_DATASETS.md(4KB)
----README.md(4KB)
----utils()
--------my_data_parallel.py(8KB)
--------misc.py(18KB)
--------f_boundary.py(7KB)
--------__init__.py(0B)
--------trnval_utils.py(12KB)
--------attr_dict.py(2KB)
--------results_page.py(8KB)
----config.py(12KB)
----.runx(292B)
----loss()
--------optimizer.py(6KB)
--------utils.py(13KB)
--------rmi_utils.py(5KB)
--------rmi.py(9KB)
--------radam.py(4KB)
----train.py(25KB)
----LICENSE(1KB)
----transforms()
--------__init__.py(0B)
--------joint_transforms.py(25KB)
--------transforms.py(12KB)
----datasets()
--------base_loader.py(9KB)
--------uniform.py(12KB)
--------cityscapes_labels.py(12KB)
--------mapillary.py(4KB)
--------randaugment.py(7KB)
--------utils.py(645B)
--------__init__.py(8KB)
--------cityscapes.py(9KB)
--------nullloader.py(3KB)
--------sampler.py(4KB)
----network()
--------mynn.py(3KB)
--------bn_helper.py(350B)
--------Resnet.py(8KB)
--------SEresnext.py(15KB)
--------deeper.py(4KB)
--------xception.py(11KB)
--------ocr_utils.py(6KB)
--------ocrnet.py(12KB)
--------utils.py(13KB)
--------attnscale.py(14KB)
--------mscale.py(18KB)
--------__init__.py(1KB)
--------wider_resnet.py(16KB)
--------mscale2.py(10KB)
--------deepv3.py(6KB)
--------hrnetv2.py(18KB)
--------basic.py(4KB)
----imgs()
--------test_imgs()
--------composited_sf.png(1.9MB)
----scripts()
--------dump_folder.yml(536B)
--------dump_cityscapes.yml(497B)
--------train_mapillary.yml(504B)
--------eval_mapillary.yml(482B)
--------train_cityscapes_sota.yml(842B)
--------train_cityscapes_deepv3.yml(426B)
--------eval_cityscapes.yml(433B)
--------train_cityscapes.yml(634B)
----requirements.txt(76B)