jimmyzzu / lung_diagnostics

Keras implementation of Noise Masking RNN for respiratory sound classification

Lung Diagnostics

This repository is an implementation of Noise Masking RNN for respiratory sound classification proposed in our paper

Video presentation about the project

It includes a preprocessing functions to convert raw .wav respiratory sound files and train script

For training

Download model and data dirs from here

Run python3 train.py --gpu GPU_NUM --data_path DATA_PATH --cv_path CV_SPLIT_PATH --exp_path EXPERIMENT_PATH

(data and cv split are included here in /data directory)

For testing

Run python3 predict.py --wav WAVFILE_PATH

Script prints text: "Probability of anomalies: X%"