README.md

metadata

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:

from transformers import AutoModel

model = AutoModel.from_pretrained("AndyWu0719/unet2dattentionmodel")

## Model Details
Model type: UNet 2D Attention
Task: MRI Tumor Segmentation
Framework: Tensorflow