min121101/sia-unet

A Unet with Sequence Information for Gastrointestinal Tract Segmentation

SIA-Unet: A Unet with Sequence Informationfor Gastrointestinal Tract Segmentation

---

The overview of the SIA-Unet.

This project is a pytorch implementation of SIA-Unet: A Unet with Sequence Information for Gastrointestinal Tract Segmentation, and this paper had accepted in PRICAI'2022.

Installation

---

Requirements

• Linux or macOS with Python = 3.9
• PyTorch = 1.10.0 and torchvision that matches the PyTorch installation. Install them together at pytorch.org to make sure of this
• albumentations package
• segmentation_model_pytorch package

You can get those package with the following line of code.

pip install albumentations

---

Dataset

---

You can get the dataset with the following line of code.

kaggle competitions download -c uw-madison-gi-tract-image-segmentation

---

Training

---

you can get the model you need by the following code

---

Att_Unet model:

cd .. #your project path

python main.py --cfg Att_Unet.yaml

---

scSE_Unet model:

cd .. #your project path

python main.py --cfg scSE_Unet.yaml

---

SIA_Unet model:

cd .. #your project path

python main.py --cfg SIA_Unet.yaml

---

Unet++ model:

cd .. #your project path

python main.py --cfg Unet++.yaml

---

Unet model:

cd .. #your project path

python main.py --cfg Unet.yaml