KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #ai-agents #dsh-plugin #codex #developer-tools #claude-code #windows #desktop-app #typescript #rust #ai-agent #cli

AirMorph

★ 29 Open GitHub ↗

An end-to-end framework for pulmonary airway analysis

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# AirwayNet

[![Python](https://img.shields.io/badge/Python-3.9%2B-blue)](https://www.python.org/)
[![PyTorch](https://img.shields.io/badge/PyTorch-2.5%2B-red)](https://pytorch.org/)

`AirwayNet` is a robust, end-to-end deep learning pipeline enabling fully automatic and comprehensive airway anatomical labeling at lobar, segmental, and subsegmental resolutions. To facilitate precise clinical interpretation, we further propose an anatomical signature, quantifying critical morphological airway features—including stenosis, ectasia, tortuosity, divergence, length, and complexity. Additionally, AirwayNet supports efficient automated branching pattern analysis, significantly enhancing bronchoscopic navigation planning and procedural safety.

- [System Requirements](#system-requirements)
- [Installation Guide](#installation-guide)
- [Instructions for use and Demo](#instructions-for-use-and-demo)

# System Requirements
## Hardware requirements
AirwayNet was deployed and rigorously validated on a Linux workstation operating on Ubuntu 20.04. The hardware configuration includes a 12th Gen Intel® Core™ i9-12900KF CPU, 64 GB system memory, and an NVIDIA RTX 3090 GPU with 24 GB of VRAM.

## Software requirements
### OS Requirements
+ Linux: Ubuntu 20.04

### Python Dependencies
`AirwayNet` mainly depends on the Python scientific stack.

```
torch
pycuda
monai
numpy
SimpleITK
... ...
```

The full packages with specific version is included in `requirements.txt`

# Installation Guide
## Install from Github and with Anaconda
```
git clone https://github.com/EndoluminalSurgicalVision-IMR/AirMorph
cd AirMorph
```

Create the virtual environment with [Anaconda](https://www.anaconda.com/)
```
conda create -n airwayatlas python=3.10
conda activate airwayatlas
```

Install PyTorch based on your system configuration. For example:
```
pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124
```

Then, install the remaining dependencies:
```
pip install -r requirements.txt
```

# Instructions for use and Demo
## Quick Start
We provide a dedicated pipeline wrapper in `airwayatlas_pipeline.py`, which utilizes the `sample_data` directory as the default input folder. For a quick start, simply execute the following command:

```
python airwayatlas_pipeline.py
```

This will run the pipeline, enabling users to quickly test and explore the functionality of AirwayNet.

## Step by Step Instruction
For comprehensive guidance on utilizing the software and analyzing the output, please refer to the detailed instructions provided in [Detailed_Instruction.md](Instruction_for_use/README.md).

### Data
The data is provided in `/sample_data`, which include the patient chest CT scan. AirwayNet processes these CT scans as input to generate binary airway models and hierarchical anatomical labels.

### AirwayNet Process
The dicated pipeline wrapper in `airwayatlas_pipeline.py`, simply execute the following command:

```
python airwayatlas_pipeline.py
```

You will see similar log messages as follows:

```
(INFO) AirwayNet: Binary Airway initialized.
(INFO) AirwayNet: Multi-class Airway Anatomy initialized. 
(INFO) AirwayNet: Lobe extraction complete.
(INFO) AirwayNet: Airway modeling complete.
(INFO) AirwayNet: Skeleton computation complete.
(INFO) AirwayNet: Saving complete.
(INFO) AirwayNet: Finished tree-parsing.
(INFO) AirwayNet: Finished feature extraction.
(INFO) AirwayNet: Finished inference dataset building.
(INFO) AirwayNet: Finished prediction.
```

### Data input and output structure
Data input structure:

```
project_root/
├── sample_data/              # Input directory containing patient chest CT scans
│   ├── patient_01/
│   │   ├── image.nii.gz      # CT scan file
```

After the AirwayNet processing is complete, the main results are, by default, saved in the same directory as the input data:

```
project_root/
├── sample_data/              # Input directory containing patient chest CT scans
│   ├── patient_01/
│   │   ├── image.nii.gz                             # CT scan file
│   │   ├── airway_bin.nii.gz                        # binary airway structure
│   │   ├── patient_01_pred_lob.nii.gz               # lobar airway anatomy
│   │   ├── patient_01_pred_seg.nii.gz               # segmental airway anatomy
│   │   ├── patient_01_pred_sub.nii.gz               # subsegmental airway anatomy
```

Meanwhile, the efficient Branching Pattern Analysis can be found in ```branchingpattern/airwaybranchpattern_pipeline.py```. The morphological airway signatures can be found in ```features/airway_morph_features.py```. 
These are described in detail in the `Method section` with accompanying pseudocode.

### Expected run time
AirwayNet was deployed and tested on a system with the following hardware configuration: a 12th Gen Intel® Core™ i9-12900KF CPU, 64 GB of system memory, and an NVIDIA RTX 3090 GPU with 24 GB of VRAM.

The pipeline processes relatively large 3D CT scans (with a typical volume size of approximately 700×512×512) as input and requires about 5–7 minutes per case to complete the entire process.

# Gallery of Results
### AirwayNet: Automated Technical Pipeline
<div align=center><img src="figs/example1.png"></div>

### AirwayNet: Unified and Fine-grained Evaluation
<div align=center><img src="figs/example3.png"></div>

### AirwayNet: Efficient Branching Pattern Analysis
<div align=center><img src="figs/example4.png"></div>

### AirwayNet: Anatomical Signatures with Clinical Relevance
<div align=center><img src="figs/example2.png"></div>

<!-- > By the Team of Institute of Medical Robotics, Shanghai Jiao Tong University, Shanghai, China

## Introduction
>> In this work, we introduce AirwayNet, a robust, end-to-end deep learning pipeline enabling fully automatic and comprehensive airway anatomical labeling at lobar, segmental, and subsegmental resolutions. To facilitate precise clinical interpretation, we further propose an anatomical signature, quantifying critical morphological airway features—including stenosis, ectasia, tortuosity, divergence, length, and complexity. Additionally, AirwayNet supports efficient automated branching pattern analysis, significantly enhancing bronchoscopic navigation planning and procedural safety.

## Usage
<div align=center><img src="figs/example3.png"></div>

### Binary Airway Modeling
Please refer to ```segmentator/airway_segmentator.py```.

### Airway Anatomical Modeling
Please refer to ```classifier/airway_classifier.py```.

Optionally, you can use the script for a quick start:

```
python airwayatlas_pipeline.py
```

### Airway Signature

The morphological airway signatures can be found in ```features/airway_morph_features.py```.

Optionally, you can use the script for a quick start:

```
python airwaysign_pipeline.py
```

### Airway BranchingPattern

Please refer to ```branchingpattern/airwaybranchpattern_pipeline.py```.

### Pretraind Model
The pretrained model could be accessed by this [link](https://drive.google.com/drive/folders/1T6VwUnHSkWzL7ghkImbWTqk6SGB-pan-?usp=sharing)

### Sample Data
The sample data could be accessed by this [link](https://drive.google.com/drive/folders/1CvkkL_EP1QcgvKiNIt7I_Yypij1ibflq?usp=sharing)

### Full Paper
More details and results of AirwayNet can be accessed by this [link](https://arxiv.org/abs/2412.11039)

## Citation
If you find this repository or our paper useful, please consider citing our paper:

```bibTex
@article{zhang2024digitalized,
  title={AirMorph: Topology-Preserving Deep Learning for Pulmonary Airway Analysis},
  author={Zhang, Minghui and Li, Chenyu and Zhang, Hanxiao and Liu, Yaoyu and Gu, Yun},
  journal={arXiv preprint arXiv:2412.11039},
  year={2024}
}
``` -->

AirMorph

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README