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.
Fuente del modelo
- Tipo
- Checkpoint, Segmentation model, UNet
- Tamaño del archivo
- 1,17 GB
- Versión
- vRevision 63cc6b8
- Actualizado
- 3 oct 2026
- Publicador
- @gplazaarriola
- Descargas
- 0
- Tareas
- Segmentación de imágenes
- Fuente
- Hugging Face
Descripción de la fuente
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)
Fuentes
1 fuente
Fuentes
1 fuenteVerificado 3 oct
Artefactos del modelo
1 artefacto
Artefactos del modelo
1 artefactoExtractos de fuentes
3 extractos
Extractos de fuentes
3 extractos--- 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
