Skip to content

LiuLiluZJU/AI-Co-Pilot-Bronchoscope-Robot

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

30 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AI Co-pilot Bronchoscope Robot — Official PyTorch Implementation

by Jingyu Zhang, Lilu Liu, Pingyu Xiang, Qin Fang, Xiuping Nie, Honghai Ma, Jian Hu, Rong Xiong, Yue Wang, Haojian Lu.

Video: AI Co-pilot Bronchoscope Robot

Introduction

Bronchoscopy is a critical diagnostic and therapeutic tool in managing lung diseases. Nevertheless, the unequal distribution of medical resources confines access to bronchoscopy primarily to well-equipped hospitals in developed regions. The scarcity of experienced practitioners contributes to the unavailability of bronchoscopic services in underdeveloped areas. In response, we present an artificial intelligence (AI) co-pilot bronchoscope robot that empowers novice doctors to conduct lung examinations as safely and adeptly as experienced colleagues. The system features a user-friendly, plug-and-play catheter with diameters of 3.3 mm or 2.1 mm, devised for robot-assisted steering, facilitating access to bronchi beyond the fifth generation in average adult patients. Drawing upon historical bronchoscopic videos and expert imitation, our AI-human shared control algorithm enables novice doctors to attain safe steering along the central axis of the bronchus, mitigating misoperations and reducing contact between the bronchoscope and bronchial walls. To evaluate the system’s reliable steering performance, we conducted in vitro assessments using realistic human bronchial phantoms simulating human respiratory behavior, and preclinical in vivo tests in live porcine lungs. Both in vitro and in vivo results underscore that our system equips novice doctors with the skills to perform lung examinations as expertly as seasoned practitioners. This study carries the potential to not only augment the existing bronchoscopy medical paradigm but also provide innovative strategies to tackle the pressing issue of medical resource disparities through AI assistance.

avatar

Usage

Prerequisites

  • Python 3.6.2
  • PyTorch 1.9.1 and torchvison (https://pytorch.org/)
  • VTK 8.2.0
  • Pyrender 0.1.45
  • PyBullet 3.2.2
  • CUDA 10.2

Installation

Necessary Python packages can be installed by

pip install -r requirements.txt

Train

> python train.py  --dataset-dir YOUR_AIRWAY_AND_CENTERLINE_DIR

The training dataset is not avaiable because of the privacy protection.

Test

The testing environment of Patient 3 and network model trained on Patient 1 and 2 can be downloaded in Google Drive. Save /airways and /checkpoints folders in root direction of the repo, then type the following code for evaulating:

> python test.py --human

Visualization and interaction GUI in PyBullet: avatar User provide discrete huamn commands (left, right, up, down, forward, i.e., H, L, I, K on keyboard) in PyBullet GUI to control the simulated bronchocope robot to safely instert in bronchus.

Besides, the automatic control by our Artificial Expert Agent can be implemented by simply typing:

> python test.py

abc

Results

Image translation (structure-preserving domain adaptation)

avatar

Simulation

avatar avatar

In-vitro

avatar

In-vivo

avatar

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages