OCULARNet
We present the OCULARNet model, a base UNet encoder - (256, 128, 64, 32) layer size - coupled with a RepVGG-B3 decoder loaded from segmentationmodelspytorch for retinal artery-vein segmentation.
Model source
- Kind
- Checkpoint, Segmentation model, UNet
- File size
- 1.17 GB
- Version
- vRevision 63cc6b8
- Updated
- Oct 3, 2026
- Publisher
- @gplazaarriola
- Downloads
- 0
- Tasks
- Image segmentation
- Source
- Hugging Face
Source description
Retinal Artery-Vein Segmentation Models
Available checkpoints
We present the OCULARNet model, a base UNet encoder - (256, 128, 64, 32) layer size - coupled with a RepVGG-B3 decoder loaded from segmentation_models_pytorch for retinal artery-vein segmentation.
Datasets, training and inference details can be found in our GitHub repository.
Usage
import torch
device = torch.device('cuda:0' if use_cuda else 'cpu')
checkpoint = torch.load("OCULARNet.pth", map_location=device)
if "model_state_dict" in checkpoint:
state_dict = checkpoint["model_state_dict"]
else:
state_dict = checkpoint
model.load_state_dict(state_dict)
Sources
1 source
Sources
1 sourceVerified Oct 3
Model artifacts
1 artifact
Model artifacts
1 artifactSource excerpts
3 excerpts
Source excerpts
3 excerpts--- license: mit pipeline_tag: image-segmentation --- # Retinal Artery-Vein Segmentation Models ## Available checkpoints We present the OCULARNet model, a base UNet encoder - (256, 128, 64, 32) layer size - coupled with a RepVGG-B3 decoder loaded from `segmentation_models_pytorch` for retinal artery-vein segmentation. Datasets, training and inference details can be found in our [GitHub repository](`https://github.com/GonzaloPlaaza/OCULAR`). ## Usage ```python import torch device = torch.device('cuda:0' if use_cuda else 'cpu') checkpoint = torch.load("OCULARNet.pth", map_location=device) if "model_state_dict" in checkpoint: state_dict = checkpoint["model_state_dict"] else: state_dict = checkpoint model.load_state_dict(state_dict)
Source context: 0 downloads · 0 likes · Pipeline image-segmentation · Repo Anon-User-Retina/OCULARNet
Source context: 0 downloads · 0 likes · Pipeline image-segmentation · Repo gplazaarriola/OCULARNet
