├── .gitignore ├── README.md ├── data ├── 0006R0_f00930.png └── 0006R0_f00930.xml ├── doc ├── ade20k_dataset.md ├── camvid_dataset.md ├── cityscapes.md ├── deeplab_understanding.md ├── develop_ideas.md ├── drn_understanding.md ├── duc_understanding.md ├── enet_understanding.md ├── fcdensenet_understanding.md ├── fcn_understanding.md ├── mapillary_vistas_dataset.md ├── pascal_voc_dataset.md ├── pspnet_understanding.md ├── refinenet_understanging.md ├── unet_understanding.md └── visdom_problem.md ├── misc ├── camvid_bbox.py ├── camvid_bbox_rename.py ├── cityscapes_bbox.py ├── loss_smooth.py ├── miou_expand.py ├── split_dataset_train_val.py ├── squeeze-segnet.py ├── transform.py └── visdom_offline_data.py ├── performance_table.py ├── semseg ├── __init__.py ├── caffe_pb2.py ├── dataloader │ ├── __init__.py │ ├── ade20k_loader.py │ ├── camvid_loader.py │ ├── camvid_lrn_loader.py │ ├── cityscapes_loader.py │ ├── folder2lmdb.py │ ├── freespace_loader.py │ ├── freespacepred_loader.py │ ├── movingmnist_loader.py │ ├── segmpred_loader.py │ ├── tfrecords_loader.py │ ├── utils.py │ └── yolodataset_loader.py ├── loss.py ├── metrics.py ├── modelloader │ ├── EDANet.py │ ├── __init__.py │ ├── bisenet.py │ ├── deconvnet.py │ ├── deeplab_resnet.py │ ├── deeplabv3.py │ ├── drn.py │ ├── drn_a_irb.py │ ├── drn_a_mt.py │ ├── drn_a_refine.py │ ├── drn_pred.py │ ├── duc_hdc.py │ ├── enet.py │ ├── enetv2.py │ ├── erfnet.py │ ├── fast_segnet.py │ ├── fc_densenet.py │ ├── fcn.py │ ├── fcn_mobilenet.py │ ├── fcn_resnet.py │ ├── fcn_shufflenet.py │ ├── frrn.py │ ├── gcn.py │ ├── lrn.py │ ├── pspnet.py │ ├── segnet.py │ ├── segnet_unet.py │ ├── sqnet.py │ ├── unet.py │ └── utils.py ├── netloader │ ├── __init__.py │ ├── resnet.py │ ├── resnet_ibn_a.py │ └── resnet_ibn_b.py ├── pytorch_modelsize.py ├── schedulers.py ├── utils │ ├── __init__.py │ ├── flops_benchmark.py │ ├── get_class_weights.py │ └── model_info_eval.py ├── visualize.py └── yoloLoss.py ├── train.py ├── train_lrn.py ├── train_mt.py ├── train_pred.py ├── validate.py ├── validate_mt.py ├── validate_pred.py └── visualize_test.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/README.md -------------------------------------------------------------------------------- /data/0006R0_f00930.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/data/0006R0_f00930.png -------------------------------------------------------------------------------- /data/0006R0_f00930.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/data/0006R0_f00930.xml -------------------------------------------------------------------------------- /doc/ade20k_dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/ade20k_dataset.md -------------------------------------------------------------------------------- /doc/camvid_dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/camvid_dataset.md -------------------------------------------------------------------------------- /doc/cityscapes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/cityscapes.md -------------------------------------------------------------------------------- /doc/deeplab_understanding.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/deeplab_understanding.md -------------------------------------------------------------------------------- /doc/develop_ideas.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/develop_ideas.md -------------------------------------------------------------------------------- /doc/drn_understanding.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/drn_understanding.md -------------------------------------------------------------------------------- /doc/duc_understanding.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/duc_understanding.md -------------------------------------------------------------------------------- /doc/enet_understanding.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/enet_understanding.md -------------------------------------------------------------------------------- /doc/fcdensenet_understanding.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/fcdensenet_understanding.md -------------------------------------------------------------------------------- /doc/fcn_understanding.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/fcn_understanding.md -------------------------------------------------------------------------------- /doc/mapillary_vistas_dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/mapillary_vistas_dataset.md -------------------------------------------------------------------------------- /doc/pascal_voc_dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/pascal_voc_dataset.md -------------------------------------------------------------------------------- /doc/pspnet_understanding.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/pspnet_understanding.md -------------------------------------------------------------------------------- /doc/refinenet_understanging.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/refinenet_understanging.md -------------------------------------------------------------------------------- /doc/unet_understanding.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/unet_understanding.md -------------------------------------------------------------------------------- /doc/visdom_problem.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/doc/visdom_problem.md -------------------------------------------------------------------------------- /misc/camvid_bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/misc/camvid_bbox.py -------------------------------------------------------------------------------- /misc/camvid_bbox_rename.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/misc/camvid_bbox_rename.py -------------------------------------------------------------------------------- /misc/cityscapes_bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/misc/cityscapes_bbox.py -------------------------------------------------------------------------------- /misc/loss_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/misc/loss_smooth.py -------------------------------------------------------------------------------- /misc/miou_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/misc/miou_expand.py -------------------------------------------------------------------------------- /misc/split_dataset_train_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/misc/split_dataset_train_val.py -------------------------------------------------------------------------------- /misc/squeeze-segnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/misc/squeeze-segnet.py -------------------------------------------------------------------------------- /misc/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/misc/transform.py -------------------------------------------------------------------------------- /misc/visdom_offline_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/misc/visdom_offline_data.py -------------------------------------------------------------------------------- /performance_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/performance_table.py -------------------------------------------------------------------------------- /semseg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semseg/caffe_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/caffe_pb2.py -------------------------------------------------------------------------------- /semseg/dataloader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semseg/dataloader/ade20k_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/ade20k_loader.py -------------------------------------------------------------------------------- /semseg/dataloader/camvid_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/camvid_loader.py -------------------------------------------------------------------------------- /semseg/dataloader/camvid_lrn_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/camvid_lrn_loader.py -------------------------------------------------------------------------------- /semseg/dataloader/cityscapes_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/cityscapes_loader.py -------------------------------------------------------------------------------- /semseg/dataloader/folder2lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/folder2lmdb.py -------------------------------------------------------------------------------- /semseg/dataloader/freespace_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/freespace_loader.py -------------------------------------------------------------------------------- /semseg/dataloader/freespacepred_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/freespacepred_loader.py -------------------------------------------------------------------------------- /semseg/dataloader/movingmnist_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/movingmnist_loader.py -------------------------------------------------------------------------------- /semseg/dataloader/segmpred_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/segmpred_loader.py -------------------------------------------------------------------------------- /semseg/dataloader/tfrecords_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/tfrecords_loader.py -------------------------------------------------------------------------------- /semseg/dataloader/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/utils.py -------------------------------------------------------------------------------- /semseg/dataloader/yolodataset_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/dataloader/yolodataset_loader.py -------------------------------------------------------------------------------- /semseg/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/loss.py -------------------------------------------------------------------------------- /semseg/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/metrics.py -------------------------------------------------------------------------------- /semseg/modelloader/EDANet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/EDANet.py -------------------------------------------------------------------------------- /semseg/modelloader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semseg/modelloader/bisenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/bisenet.py -------------------------------------------------------------------------------- /semseg/modelloader/deconvnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/deconvnet.py -------------------------------------------------------------------------------- /semseg/modelloader/deeplab_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/deeplab_resnet.py -------------------------------------------------------------------------------- /semseg/modelloader/deeplabv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/deeplabv3.py -------------------------------------------------------------------------------- /semseg/modelloader/drn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/drn.py -------------------------------------------------------------------------------- /semseg/modelloader/drn_a_irb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/drn_a_irb.py -------------------------------------------------------------------------------- /semseg/modelloader/drn_a_mt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/drn_a_mt.py -------------------------------------------------------------------------------- /semseg/modelloader/drn_a_refine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/drn_a_refine.py -------------------------------------------------------------------------------- /semseg/modelloader/drn_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/drn_pred.py -------------------------------------------------------------------------------- /semseg/modelloader/duc_hdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/duc_hdc.py -------------------------------------------------------------------------------- /semseg/modelloader/enet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/enet.py -------------------------------------------------------------------------------- /semseg/modelloader/enetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/enetv2.py -------------------------------------------------------------------------------- /semseg/modelloader/erfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/erfnet.py -------------------------------------------------------------------------------- /semseg/modelloader/fast_segnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/fast_segnet.py -------------------------------------------------------------------------------- /semseg/modelloader/fc_densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/fc_densenet.py -------------------------------------------------------------------------------- /semseg/modelloader/fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/fcn.py -------------------------------------------------------------------------------- /semseg/modelloader/fcn_mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/fcn_mobilenet.py -------------------------------------------------------------------------------- /semseg/modelloader/fcn_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/fcn_resnet.py -------------------------------------------------------------------------------- /semseg/modelloader/fcn_shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/fcn_shufflenet.py -------------------------------------------------------------------------------- /semseg/modelloader/frrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/frrn.py -------------------------------------------------------------------------------- /semseg/modelloader/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/gcn.py -------------------------------------------------------------------------------- /semseg/modelloader/lrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/lrn.py -------------------------------------------------------------------------------- /semseg/modelloader/pspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/pspnet.py -------------------------------------------------------------------------------- /semseg/modelloader/segnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/segnet.py -------------------------------------------------------------------------------- /semseg/modelloader/segnet_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/segnet_unet.py -------------------------------------------------------------------------------- /semseg/modelloader/sqnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/sqnet.py -------------------------------------------------------------------------------- /semseg/modelloader/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/unet.py -------------------------------------------------------------------------------- /semseg/modelloader/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/modelloader/utils.py -------------------------------------------------------------------------------- /semseg/netloader/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python 2 | # -*- coding: UTF-8 -*- 3 | -------------------------------------------------------------------------------- /semseg/netloader/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/netloader/resnet.py -------------------------------------------------------------------------------- /semseg/netloader/resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/netloader/resnet_ibn_a.py -------------------------------------------------------------------------------- /semseg/netloader/resnet_ibn_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/netloader/resnet_ibn_b.py -------------------------------------------------------------------------------- /semseg/pytorch_modelsize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/pytorch_modelsize.py -------------------------------------------------------------------------------- /semseg/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/schedulers.py -------------------------------------------------------------------------------- /semseg/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | -------------------------------------------------------------------------------- /semseg/utils/flops_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/utils/flops_benchmark.py -------------------------------------------------------------------------------- /semseg/utils/get_class_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/utils/get_class_weights.py -------------------------------------------------------------------------------- /semseg/utils/model_info_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/utils/model_info_eval.py -------------------------------------------------------------------------------- /semseg/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/visualize.py -------------------------------------------------------------------------------- /semseg/yoloLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/semseg/yoloLoss.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/train.py -------------------------------------------------------------------------------- /train_lrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/train_lrn.py -------------------------------------------------------------------------------- /train_mt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/train_mt.py -------------------------------------------------------------------------------- /train_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/train_pred.py -------------------------------------------------------------------------------- /validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/validate.py -------------------------------------------------------------------------------- /validate_mt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/validate_mt.py -------------------------------------------------------------------------------- /validate_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/validate_pred.py -------------------------------------------------------------------------------- /visualize_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanfuchen/semseg/HEAD/visualize_test.py --------------------------------------------------------------------------------