segmentation-models-pytorch
Package — Deep LearningPython 3.7+Advanced
Neural network architectures for image segmentation
Quick Info
- Documentation
- Official Docs
- Python Version
- 3.7+
- Dependencies
- PyTorch, torchvision, Pillow, tqdm, pretrainedmodels, efficientnet-pytorch, timm
- Install
pip install segmentation-models-pytorch
Learn by Difficulty
Quick Example
python
# Install: pip install segmentation-models-pytorch import segmentation_models_pytorch # Basic segmentation-models-pytorch usage print(f"Using segmentation-models-pytorch") # See documentation for detailed examples
segmentation-models-pytorch is a third-party package. Neural network architectures for image segmentation. Install with: pip install segmentation-models-pytorch
Try in PlaygroundTags
packagedeep-learningneural-networkaigpu