Downloads · 30 days
5
19% of all-time downloads
Segformer_OCT_Retina
coralexbadea/Segformer_OCT_Retina
Segformer_OCT_Retina is a machine learning model from coralexbadea. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
SegFormer model fine-tuned on AROI dataset AROI: Annotated Retinal OCT Images Database.
All-time downloads
26
Public
Repo size
59.7 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt29.8 MB · 64%
At a glance
- Library
- transformers
- License
- mit
- Model type
- segformer
- Access
- Public
- Created
- Feb 6, 2024
- Updated
- Feb 6, 2024
- SHA
- 90e399ba
Model card
From the Hugging Face model README
SegFormer model fine-tuned on AROI
SegFormer model fine-tuned on AROI dataset AROI: Annotated Retinal OCT Images Database.
Disclaimer: The team releasing SegFormer did not write a model card for this model so this model card has been written by the Hugging Face team.
Model description
SegFormer consists of a hierarchical Transformer encoder and a lightweight all-MLP decode head to achieve great results on semantic segmentation benchmarks such as ADE20K and Cityscapes. The hierarchical Transformer is first pre-trained on ImageNet-1k, after which a decode head is added and fine-tuned altogether on a downstream dataset.
- Library
- transformers
- Type
- segformer
- License
- mit
- Created
- Feb 6, 2024
- Updated
- Feb 6, 2024