UAE

KO | EN

gitlite — search

#javascript #python #hacktoberfest #react #ai #typescript #llm #go #golang #android #machine-learning #rust #deep-learning #linux

★ 21 Open GitHub ↗

Implementation of "Abnormality Detection in Chest X-ray Images Using Uncertainty Prediction Autoencoders" in MICCAI 2020.

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# UAE
Implementation of "Abnormality Detection in Chest X-ray Images Using Uncertainty Prediction Autoencoders" in MICCAI 2020. 
 \[[paper](https://link.springer.com/chapter/10.1007/978-3-030-59725-2_51)\]

## Datasets
RSNA Pneumonia Detection Challenge Dataset: https://www.kaggle.com/c/rsna-pneumonia-detection-challenge/data

Pediatric Dataset: https://data.mendeley.com/datasets/rscbjbr9sj/3

## Requirements
* python, pytorch, numpy, scikit-learn, tensorboard-pytorch

## Basic usage
Before running the scripts, you should edit `DATA_PATH` in `xray_data.py` in order to read the data correctly:

```python
DATA_PATH = '$WHERE_YOU_STORE_DATASETS'
```

To train a traditional autoencoder:
```python
python uae_main.py
```
To train an autoencoder with uncertainty:
```python
python uae_main.py --u
```
To evaluate a model:
```python
python uae_main.py --eval --u(optional)
```

## Cite
```latex
@inproceedings{mao2020abnormality,
  title={Abnormality Detection in Chest X-Ray Images Using Uncertainty Prediction Autoencoders},
  author={Mao, Yifan and Xue, Fei-Fei and Wang, Ruixuan and Zhang, Jianguo and Zheng, Wei-Shi and Liu, Hongmei},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={529--538},
  year={2020},
  organization={Springer}
}
```

UAE

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

UAE

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README