This model is a lightweight U-Net trained on the Synapse multi-organ CT dataset.
It is designed to run on low-resource hardware (GTX 1650, 4GB VRAM) and CPU.
1model = UNet(n_channels=1, n_classes=9)
2model.load_state_dict(torch.load("best_model.pth"))
Synapse Multi-Organ Segmentation Challenge.
License: Review original dataset license.
