|
|
--- |
|
|
language: en |
|
|
tags: |
|
|
- image-segmentation |
|
|
- unet |
|
|
- attention |
|
|
license: apache-2.0 |
|
|
datasets: |
|
|
- BraTS |
|
|
metrics: |
|
|
- dice |
|
|
- precision |
|
|
- sensitivity |
|
|
- specificity |
|
|
pipeline_tag: image-segmentation |
|
|
source: |
|
|
- url: https://huggingface.co/spaces/AndyWu0719/unet2dattention |
|
|
--- |
|
|
|
|
|
# Space Path |
|
|
|
|
|
If you want to try this model online, you can go space path: AndyWu0719/unet2dattention |
|
|
|
|
|
## UNet 2D Attention Model for MRI Tumor Segmentation |
|
|
|
|
|
This repository contains a UNet 2D Attention model for MRI tumor segmentation. |
|
|
|
|
|
### Usage |
|
|
|
|
|
To use this model, you can load it with the `transformers` library: |
|
|
|
|
|
```python |
|
|
from transformers import AutoModel |
|
|
|
|
|
model = AutoModel.from_pretrained("AndyWu0719/unet2dattentionmodel") |
|
|
|
|
|
## Model Details |
|
|
Model type: UNet 2D Attention |
|
|
Task: MRI Tumor Segmentation |
|
|
Framework: Tensorflow |
