File size: 768 Bytes
71d2e1f 7daa53f 52dee18 d8530c5 52dee18 d8530c5 52dee18 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | ---
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)
|