detect_cough.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
""" Python wraper to detect cough """

import os
import sys
sys.path.append(os.path.abspath('./src'))
sys.path.append(os.path.abspath('./src/cough_detection'))
from src.feature_class import features
from src.DSP import classify_cough
from scipy.io import wavfile
import pickle
import argparse


def main(input_file):
    """
    Detect cough in a given audio file
    Inputs:
        input_file: (str) path to audio file
    Outputs:
        result: (float) probability that a given file is a cough
    """
    # data_folder = './sample_recordings'
    model = pickle.load(open(os.path.join('./models', 'cough_classifier'),
        'rb'))
    scaler = pickle.load(open(os.path.join('./models',
        'cough_classification_scaler'), 'rb'))

    fs, x = wavfile.read(input_file)
    prob = classify_cough(x, fs, model, scaler)
    print(f"{input_file} has probability of cough: {prob}")
    return prob

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-i', '--input',
                        help='Path to input audio file',
                        required=True)
    args = parser.parse_args()
    main(args.input)