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