ternaus/TernausNet

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

1063 stars · 244 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

TernausNet is a PyTorch implementation of the U-Net architecture with a VGG11 encoder pre-trained on ImageNet for image segmentation. It was part of the winning solution of the Kaggle Carvana Image Masking Challenge and ships with pre-trained weights via pip.

Use cases

• segment images with a pretrained UNet in PyTorch
• train a binary image segmentation model on a small dataset
• segment medical or satellite imagery
• fine-tune a UNet with an ImageNet-pretrained encoder
• reproduce a Kaggle Carvana image masking solution

When to choose

• you need a proven UNet variant with pretrained encoder weights for segmentation
• you have limited training data and want faster convergence
• you want a simple pip-installable PyTorch segmentation model

When to avoid

• you need multi-class or modern architectures like DeepLab or SegFormer
• you need a actively developed library with frequent updates
• you work outside PyTorch

Facets

library · maturity maintenance

machine-learning deep-learning computer-vision image-processing computer-vision deep-learning machine-learning image-processing python unet image-segmentation pytorch pretrained-encoder vgg11 semantic-segmentation

Member repositories

For agents

markdown · JSON · MCP: product_card(name="ternaus/TernausNet")