Esempio n. 1
0
def plot_result(wav, word_results, params, min_params, colors, items):
    file = Plotter("DAF")
    file.add_sub_plot_data("Digitized audio file", wav.get_one_channel_data(), x_label="Samples", y_label="Amplitude")
    for i in word_results.keys():
        file.add_line_at("Digitized audio file", list(map(lambda x: x * items, word_results[i]["starts"])), "x",
                         colors[i], lw=3)
        file.add_line_at("Digitized audio file", list(map(lambda x: x * items, word_results[i]["ends"])), "x",
                         colors[i], lw=3)
        file.sub_plot_all_horizontal(show=False, save=True)

    energy = Plotter("Energy")
    energy.add_sub_plot_data("Energy", params["energy"], x_label="Frames", y_label="Energy Value")
    energy.add_line_at("Energy", min_params["energy"], "y", color="red")
    energy.sub_plot_all_horizontal(show=False, save=True)

    mdf = Plotter("MDF")
    mdf.add_sub_plot_data("Most Dominant Frequency", params["mdf"], x_label="Frames", y_label="MDF Value")
    mdf.add_line_at("Most Dominant Frequency", min_params["mdf"], "y", color="red")
    mdf.sub_plot_all_horizontal(show=False, save=True)

    zcr = Plotter("ZCR")
    zcr.add_sub_plot_data("Zero Crossing Rate", params["zcr"], x_label="Frames", y_label="ZCR Value")
    zcr.add_line_at("Zero Crossing Rate", min_params["zcr"], "y", color="red")
    zcr.sub_plot_all_horizontal(show=False, save=True)

    zcr = Plotter("SFM")
    zcr.add_sub_plot_data("Spectral Flatness Measure", params["sfm"], x_label="Frames", y_label="SFM Value")
    zcr.add_line_at("Spectral Flatness Measure", min_params["sfm"], "y", color="red")
    zcr.sub_plot_all_horizontal(show=False, save=True)