-
Segmentation Models - https://github.com/qubvel/segmentation_models.pytorch
-
MMsegmentation - https://github.com/open-mmlab/mmsegmentation
-
torchvision - https://pytorch.org/vision/stable/models.html
- FCN w/ backbones ResNet50, ResNet101
- DeepLabV3 w/ backbones ResNet50, ResNet101, MobileNetV3-Large
- LR-ASPP w/ backbones MobileNetV3-Large
- Mask R-CNN w/ pre-trained on COCO train2017 and backbone pre-trained on ImageNet
-
Detectron2 - https://github.com/facebookresearch/detectron2
-
hugginface
mikiotada / source_segmentationmodels Goto Github PK
View Code? Open in Web Editor NEWlist of image segmentation models