Explore Models

By Task
All Models (32)Object Detection Models (19)Classification Models (7)Instance Segmentation Models (3)Semantic Segmentation Models (1)Keypoint Detection Models (2)
By Feature
Foundation Vision ModelsMultimodal Vision ModelsTop LLMs with Vision CapabilitiesImage Embedding ModelsReal Time Vision ModelsZero-Shot Detection ModelsImage Captioning ModelsImage Similarity ModelsImage Tagging ModelsVisual Question Answering ModelsOptimal for Edge Deployment

Top PyTorch Image Segmentation Models

Explore image segmentation models that use the PyTorch framework.
‍
Deploy select models (i.e. YOLOv8, CLIP) using the Roboflow Hosted API, or your own hardware using Roboflow Inference.

Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation
FastSAM

Model Size:

MB

Parameters:

636000000

Architecture:

FastSAM is an image segmentation model trained using 2% of the data in the Segment Anything Model SA-1B dataset. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation
DETIC

Model Size:

25000000.0

MB

Parameters:

Architecture:

Detic is an open source segmentation model developed by Meta Research and released in 2022. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation
YOLOv8 Instance Segmentation

Model Size:

MB

Parameters:

68200000

Architecture:

YOLO

The state-of-the-art YOLOv8 model comes with support for instance segmentation tasks. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation
YOLOv7 Instance Segmentation

Model Size:

MB

Parameters:

150000000

Architecture:

YOLO

YOLOv7 Instance Segmentation lets you perform segmentation tasks with the YOLOv7 model. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation
OneFormer

Model Size:

MB

Parameters:

219 million

Architecture:

Transformers

OneFormer is a state-of-the-art multi-task image segmentation framework that is implemented using transformers. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation
Mask RCNN

Model Size:

MB

Parameters:

Architecture:

Mask RCNN is a convolutional neural network for instance segmentation. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation
YOLOv5 Instance Segmentation

Model Size:

MB

Parameters:

Architecture:

CNN, YOLO

YOLOv5 Instance Segmentation is a version of YOLOv5 that can be used for instance segmentation tasks. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide

Deploy a computer vision model today

Join 250,000+ developers curating high quality datasets and deploying better models with Roboflow.

Get started