【文件属性】:
文件名称:segmentation_pytorch:semantic segmentation pytorch 语义分割
文件大小:26KB
文件格式:ZIP
更新时间:2021-04-07 10:22:56
pytorch unet hed pspnet deeplab-v3-plus
语义分割火炬
author is leilei
Restart this project from 2017-10-01
TODO
The latest version of the code cannot be executed and is still being updated.
环境
python: 3.6+
ubuntu16.04 or 18.04
pytorch 1.6+ (cuda10.2 docker)
tensorboard 2.0+
笔记
如果引入黑色边框,它将被视为一种类型,默认值为0!
标签值是[1,N],0是黑色边框类!
不支持分布式(NCCL)
如何使用
火车
测试
数据集详细信息
root:
images:
labels:
train.txt:
【文件预览】:
segmentation_pytorch-master
----utils()
--------util.py(1KB)
--------aug_PIL.py(11KB)
--------aug_GDAL.py(14KB)
----models()
--------pspnet.py(8KB)
--------deeplab_v3_plus.py(4KB)
--------unet.py(4KB)
--------models.py(303B)
--------hed_series()
--------spp.py(3KB)
----readmes()
--------data_aug.md(399B)
--------train_cusom.md(22B)
----main.py(4KB)
----data_augmentation.py(4KB)
----README.md(2KB)
----data()
--------test.txt(0B)
--------train.txt(0B)