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

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 »
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation

Model Size:

25000000.0

MB

Parameters:

Architecture:

Detic is an open source segmentation model developed by Meta Research and released in 2022. Learn more »
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
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 »
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation

Model Size:

MB

Parameters:

150000000

Architecture:

YOLO

YOLOv7 Instance Segmentation lets you perform segmentation tasks with the YOLOv7 model. Learn more »
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation

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 »
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
Instance Segmentation

Model Size:

MB

Parameters:

Architecture:

Mask RCNN is a convolutional neural network for instance segmentation. Learn more »
Instance Segmentation
Instance Segmentation
Deploy on Device with Roboflow✅
Instance Segmentation
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 »

Deploy a computer vision model today

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

Get started