Ir para o conteúdo
ModelVerificadoExterno
Baixar

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.

Fontehuggingface.co/Anon-User-Retina/OCULARNet/tree/63cc6b80f2ee2eceb39e3fe99d1c49cea7cec927
TipoCheckpointVersãoRevision 63cc6b8Publicador@gplazaarriolaCnota

Fonte do modelo

Tipo
Checkpoint, Segmentation model, UNet
Tamanho do arquivo
1,17 GB
Versão
vRevision 63cc6b8
Atualizado
3 de out. de 2026
Publicador
@gplazaarriola
Downloads
0
Tarefas
Segmentação de imagem
Fonte
Hugging Face

Descrição da fonte

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)

Fontes

1 fonte
Hugging FacePrimárioVersão 63cc6b80f2ee2eceb39e3fe99d1c49cea7cec927
huggingface.co/Anon-User-Retina/OCULARNet/tree/63cc6b80f2ee2eceb39e3fe99d1c49cea7cec927

Verificado 3 de out.

Artefatos de modelo

1 artefato

OCULARNet.pth

pth · 1,17 GB · SHA-256 88eec1d78ec9…df75 · Hugging Face

Baixar

Trechos de fonte

3 trechos
Readme11 de jul.huggingface.co/Anon-User-Retina/OCULARNet

--- 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)

Note11 de jul.huggingface.co/Anon-User-Retina/OCULARNet

Source context: 0 downloads · 0 likes · Pipeline image-segmentation · Repo Anon-User-Retina/OCULARNet

Note9 de jun.huggingface.co/api/models/gplazaarriola/OCULARNet

Source context: 0 downloads · 0 likes · Pipeline image-segmentation · Repo gplazaarriola/OCULARNet