├── .gitignore ├── LICENSE ├── README.md ├── VOCdevkit └── VOC2007 │ ├── ImageSets │ └── Segmentation │ │ └── README.md │ ├── JPEGImages │ └── README.md │ └── SegmentationClass │ └── README.md ├── datasets ├── JPEGImages │ └── 1.jpg ├── SegmentationClass │ └── 1.png └── before │ ├── 1.jpg │ └── 1.json ├── get_miou.py ├── img └── street.jpg ├── json_to_dataset.py ├── logs └── README.MD ├── model_data └── README.md ├── nets ├── __init__.py ├── backbone.py ├── segformer.py └── segformer_training.py ├── predict.py ├── requirements.txt ├── segformer.py ├── summary.py ├── train.py ├── utils ├── __init__.py ├── callbacks.py ├── dataloader.py ├── utils.py ├── utils_fit.py └── utils_metrics.py ├── voc_annotation.py └── 常见问题汇总.md /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /VOCdevkit/VOC2007/ImageSets/Segmentation/README.md: -------------------------------------------------------------------------------- 1 | 存放的是指向文件名称的txt 2 | 3 | -------------------------------------------------------------------------------- /VOCdevkit/VOC2007/JPEGImages/README.md: -------------------------------------------------------------------------------- 1 | 这里面存放的是训练用的图片文件。 2 | -------------------------------------------------------------------------------- /VOCdevkit/VOC2007/SegmentationClass/README.md: -------------------------------------------------------------------------------- 1 | 这里面存放的是训练过程中产生的权重。 2 | -------------------------------------------------------------------------------- /datasets/JPEGImages/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/datasets/JPEGImages/1.jpg -------------------------------------------------------------------------------- /datasets/SegmentationClass/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/datasets/SegmentationClass/1.png -------------------------------------------------------------------------------- /datasets/before/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/datasets/before/1.jpg -------------------------------------------------------------------------------- /datasets/before/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/datasets/before/1.json -------------------------------------------------------------------------------- /get_miou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/get_miou.py -------------------------------------------------------------------------------- /img/street.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/img/street.jpg -------------------------------------------------------------------------------- /json_to_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/json_to_dataset.py -------------------------------------------------------------------------------- /logs/README.MD: -------------------------------------------------------------------------------- 1 | 这一部分用来存放训练后的文件。 2 | This part is used to store post training documents. -------------------------------------------------------------------------------- /model_data/README.md: -------------------------------------------------------------------------------- 1 | 这里面存放的是已经训练好的权重，可通过百度网盘下载。 2 | -------------------------------------------------------------------------------- /nets/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /nets/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/nets/backbone.py -------------------------------------------------------------------------------- /nets/segformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/nets/segformer.py -------------------------------------------------------------------------------- /nets/segformer_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/nets/segformer_training.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /segformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/segformer.py -------------------------------------------------------------------------------- /summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/summary.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/utils/callbacks.py -------------------------------------------------------------------------------- /utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/utils/dataloader.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/utils_fit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/utils/utils_fit.py -------------------------------------------------------------------------------- /utils/utils_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/utils/utils_metrics.py -------------------------------------------------------------------------------- /voc_annotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/voc_annotation.py -------------------------------------------------------------------------------- /常见问题汇总.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/segformer-pytorch/HEAD/常见问题汇总.md --------------------------------------------------------------------------------