qubvel-org/segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
- Stars
- 12k
- Forks
- 1.9k
- Open issues
- 71
- Last push
- 3d ago
About segmentation_models.pytorch
segmentation_models.pytorch is a free, open-source computer vision tool by qubvel-org, written mainly in Python. Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones. It has 12k GitHub stars and 1.9k forks, and was last updated 3d ago. It's released under the MIT license, a standard open-source license.
- License
- MIT
- Language
- Python
- Stars
- 11,777
- Last update
- 3d ago
- Runs on Mac
- Not stated
- GPU
- Not stated
- Paid API key
- Not stated
- Self-hostable
- Not stated
Similar free tools
ultralytics63k stars · Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking
supervision51k stars · We write your reusable computer vision tools. 💜
vit-pytorch26k stars · Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
cvat17k stars · CVAT is a leading data annotation platform for image, video, audio, and 3D datasets. It offers open-source, cloud, and enterprise products, as well as labeling services, with AI-assisted labeling, qua
label-studio29k stars · Label Studio is a multi-type data labeling and annotation tool with standardized output format
segmentation_models.pytorch FAQ
Is segmentation_models.pytorch free?
Yes. segmentation_models.pytorch is open source and free to use. It's released under the MIT license, a standard open-source license.
Does segmentation_models.pytorch run on a Mac?
The project doesn't say. Check its installation guide before you start.
Does segmentation_models.pytorch need a GPU?
The project doesn't state hardware needs. AI models generally run much faster with a GPU.
Is segmentation_models.pytorch still maintained?
Yes. It was last updated 3d ago and has 71 open issues on GitHub.
What are alternatives to segmentation_models.pytorch?
Similar free, open-source tools include ultralytics, yolov5, supervision, vit-pytorch.