Skip to content

README.md

Latest commit

 

History

History
38 lines (30 loc) · 1.25 KB

README.md

File metadata and controls

38 lines (30 loc) · 1.25 KB

Shape-Aware Refinement

This research proposes a PDE-based trainable refinement module for tubular-structure segmentation to enhance shape continuity. image

Paper / Slides

Overview

  • PDE-based refinement improves mask continuity while maintaining thin shape
  • Our approach is applied to various architectures for tubular-structure segmentation because of post-process

Dataset

Environment

  • Ubuntu 24.04.2 LTS
  • NVIDIA TITAN RTX(Memory: 24GB) * 4

How to run

  1. setup the environment.
$ pip install -r requirements.txt

  1. Download DRIVE dataset to /dataset.

  2. Run /dataset/data_process.py to fit dataset for training models.

$ cd dataset
$ python data_process.py -dp DATASET_PATH -dn DATASET_NAME
  1. train and test model.
$ cd pde-shape-refiner
$ main_ddp.sh