U-Net Model for Image Segmentation

This is a trained U-Net model for segmentation tasks.

Usage

import torch
from unet import UNet  # Replace with actual model import
model = UNet()
model.load_state_dict(torch.load("unet.pth"))
model.eval()

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support