├── README.md ├── configs └── parameter.yaml ├── data ├── images │ ├── aaa.jpg │ ├── bbb.jpg │ └── ccc.jpg ├── labels │ ├── aaa.jpg │ ├── bbb.jpg │ └── ccc.jpg ├── test.txt └── train.txt ├── data_augmentation.py ├── demo.py ├── main.py ├── models ├── deeplab_v3_plus.py ├── hed_series │ ├── hed_res.py │ ├── hed_vgg16.py │ ├── hf_fcn_res.py │ └── hf_fcn_vgg16.py ├── models.py ├── pspnet.py ├── spp.py └── unet.py ├── readmes ├── data_aug.md ├── main_modify.jpg ├── param_modify.jpg └── train_cusom.md └── utils ├── aug_GDAL.py ├── aug_PIL.py ├── dataset.py ├── metrics.py ├── plots.py ├── trainval.py └── util.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/README.md -------------------------------------------------------------------------------- /configs/parameter.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/configs/parameter.yaml -------------------------------------------------------------------------------- /data/images/aaa.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/data/images/aaa.jpg -------------------------------------------------------------------------------- /data/images/bbb.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/data/images/bbb.jpg -------------------------------------------------------------------------------- /data/images/ccc.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/data/images/ccc.jpg -------------------------------------------------------------------------------- /data/labels/aaa.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/data/labels/aaa.jpg -------------------------------------------------------------------------------- /data/labels/bbb.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/data/labels/bbb.jpg -------------------------------------------------------------------------------- /data/labels/ccc.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/data/labels/ccc.jpg -------------------------------------------------------------------------------- /data/test.txt: -------------------------------------------------------------------------------- 1 | /home/gengyanlei/data/images/ccc.jpg -------------------------------------------------------------------------------- /data/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/data/train.txt -------------------------------------------------------------------------------- /data_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/data_augmentation.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/demo.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/main.py -------------------------------------------------------------------------------- /models/deeplab_v3_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/models/deeplab_v3_plus.py -------------------------------------------------------------------------------- /models/hed_series/hed_res.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/models/hed_series/hed_res.py -------------------------------------------------------------------------------- /models/hed_series/hed_vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/models/hed_series/hed_vgg16.py -------------------------------------------------------------------------------- /models/hed_series/hf_fcn_res.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/models/hed_series/hf_fcn_res.py -------------------------------------------------------------------------------- /models/hed_series/hf_fcn_vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/models/hed_series/hf_fcn_vgg16.py -------------------------------------------------------------------------------- /models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/models/models.py -------------------------------------------------------------------------------- /models/pspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/models/pspnet.py -------------------------------------------------------------------------------- /models/spp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/models/spp.py -------------------------------------------------------------------------------- /models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/models/unet.py -------------------------------------------------------------------------------- /readmes/data_aug.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/readmes/data_aug.md -------------------------------------------------------------------------------- /readmes/main_modify.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/readmes/main_modify.jpg -------------------------------------------------------------------------------- /readmes/param_modify.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/readmes/param_modify.jpg -------------------------------------------------------------------------------- /readmes/train_cusom.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/readmes/train_cusom.md -------------------------------------------------------------------------------- /utils/aug_GDAL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/utils/aug_GDAL.py -------------------------------------------------------------------------------- /utils/aug_PIL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/utils/aug_PIL.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/utils/trainval.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/segmentation_pytorch/HEAD/utils/util.py --------------------------------------------------------------------------------