├── .gitignore ├── README.md ├── configs ├── data.yaml └── model_yolo_segmentation.yaml ├── data ├── __init__.py ├── custom_transforms.py ├── dataloaders │ ├── __init__.py │ ├── custom_transforms.py │ ├── datasets │ │ ├── __init__.py │ │ ├── cityscapes.py │ │ ├── coco.py │ │ ├── combine_dbs.py │ │ └── pascal.py │ └── utils.py ├── transform_voc.py └── utils.py ├── doc_img ├── visualize-dataset.png └── visualize-img.png ├── images ├── berlin_000018_000019_leftImg8bit.png └── dusseldorf_000071_000019_leftImg8bit.png ├── models ├── __init__.py ├── backbone │ ├── __init__.py │ └── yolov5_seg.py ├── head │ ├── __init__.py │ └── segmentation.py ├── hub │ ├── yolov3-spp.yaml │ ├── yolov3-tiny.yaml │ ├── yolov3.yaml │ ├── yolov5-fpn.yaml │ ├── yolov5-p2.yaml │ ├── yolov5-p6.yaml │ ├── yolov5-p7.yaml │ ├── yolov5-panet.yaml │ ├── yolov5l.yaml │ ├── yolov5l6.yaml │ ├── yolov5m.yaml │ ├── yolov5m6.yaml │ ├── yolov5s.yaml │ ├── yolov5s6.yaml │ ├── yolov5s_transformer │ ├── yolov5x.yaml │ └── yolov5x6.yaml ├── loss │ ├── __init__.py │ └── yolo_seg.py ├── modules │ ├── __init__.py │ ├── activations.py │ ├── cbam.py │ ├── common.py │ ├── coord_conv.py │ ├── drop_block.py │ ├── experimental.py │ └── seblock.py ├── neck │ ├── FPN.py │ ├── PAN.py │ └── __init__.py └── seg_model.py ├── requirements.txt ├── scripts ├── train_seg.py └── visualize.py ├── train_cityscapes.sh ├── utils ├── __init__.py ├── general.py ├── lr_scheduler.py ├── metrics.py ├── saver.py └── summaries.py └── visualize.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/README.md -------------------------------------------------------------------------------- /configs/data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/configs/data.yaml -------------------------------------------------------------------------------- /configs/model_yolo_segmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/configs/model_yolo_segmentation.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/custom_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/custom_transforms.py -------------------------------------------------------------------------------- /data/dataloaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/dataloaders/__init__.py -------------------------------------------------------------------------------- /data/dataloaders/custom_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/dataloaders/custom_transforms.py -------------------------------------------------------------------------------- /data/dataloaders/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/dataloaders/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/dataloaders/datasets/cityscapes.py -------------------------------------------------------------------------------- /data/dataloaders/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/dataloaders/datasets/coco.py -------------------------------------------------------------------------------- /data/dataloaders/datasets/combine_dbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/dataloaders/datasets/combine_dbs.py -------------------------------------------------------------------------------- /data/dataloaders/datasets/pascal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/dataloaders/datasets/pascal.py -------------------------------------------------------------------------------- /data/dataloaders/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/dataloaders/utils.py -------------------------------------------------------------------------------- /data/transform_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/transform_voc.py -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/data/utils.py -------------------------------------------------------------------------------- /doc_img/visualize-dataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/doc_img/visualize-dataset.png -------------------------------------------------------------------------------- /doc_img/visualize-img.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/doc_img/visualize-img.png -------------------------------------------------------------------------------- /images/berlin_000018_000019_leftImg8bit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/images/berlin_000018_000019_leftImg8bit.png -------------------------------------------------------------------------------- /images/dusseldorf_000071_000019_leftImg8bit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/images/dusseldorf_000071_000019_leftImg8bit.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/backbone/__init__.py -------------------------------------------------------------------------------- /models/backbone/yolov5_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/backbone/yolov5_seg.py -------------------------------------------------------------------------------- /models/head/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/head/__init__.py -------------------------------------------------------------------------------- /models/head/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/head/segmentation.py -------------------------------------------------------------------------------- /models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov3-tiny.yaml -------------------------------------------------------------------------------- /models/hub/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov3.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5-p2.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5-p6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5-p7.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /models/hub/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5l.yaml -------------------------------------------------------------------------------- /models/hub/yolov5l6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5l6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5m.yaml -------------------------------------------------------------------------------- /models/hub/yolov5m6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5m6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5s.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5s6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s_transformer: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5s_transformer -------------------------------------------------------------------------------- /models/hub/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5x.yaml -------------------------------------------------------------------------------- /models/hub/yolov5x6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/hub/yolov5x6.yaml -------------------------------------------------------------------------------- /models/loss/__init__.py: -------------------------------------------------------------------------------- 1 | from .yolo_seg import SegmentationLosses -------------------------------------------------------------------------------- /models/loss/yolo_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/loss/yolo_seg.py -------------------------------------------------------------------------------- /models/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/modules/__init__.py -------------------------------------------------------------------------------- /models/modules/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/modules/activations.py -------------------------------------------------------------------------------- /models/modules/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/modules/cbam.py -------------------------------------------------------------------------------- /models/modules/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/modules/common.py -------------------------------------------------------------------------------- /models/modules/coord_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/modules/coord_conv.py -------------------------------------------------------------------------------- /models/modules/drop_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/modules/drop_block.py -------------------------------------------------------------------------------- /models/modules/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/modules/experimental.py -------------------------------------------------------------------------------- /models/modules/seblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/modules/seblock.py -------------------------------------------------------------------------------- /models/neck/FPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/neck/FPN.py -------------------------------------------------------------------------------- /models/neck/PAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/neck/PAN.py -------------------------------------------------------------------------------- /models/neck/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/neck/__init__.py -------------------------------------------------------------------------------- /models/seg_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/models/seg_model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/train_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/scripts/train_seg.py -------------------------------------------------------------------------------- /scripts/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/scripts/visualize.py -------------------------------------------------------------------------------- /train_cityscapes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/train_cityscapes.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/utils/lr_scheduler.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/utils/saver.py -------------------------------------------------------------------------------- /utils/summaries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/utils/summaries.py -------------------------------------------------------------------------------- /visualize.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Irvingao/yolov5-segmentation/HEAD/visualize.sh --------------------------------------------------------------------------------