Commit 87febe56 authored by Vladimir Malenovsky's avatar Vladimir Malenovsky
Browse files

correction of categories per talker/talker pairs in ISM1 and ISM2 tests

parent 7e6fa6c2
Loading
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -324,8 +324,8 @@ def generate_ism1_scene(
    audiofile.write(
        os.path.join(
            cfg.output_path,
            "cat"+scene["name"][1],
            cfg.use_output_prefix + os.path.basename(scene["name"]+".wav"),
            os.path.dirname(scene["name"]),
            cfg.use_output_prefix + os.path.basename(scene["name"]),
        ),
        y.audio,
        y.fs,
+11 −0
Original line number Diff line number Diff line
@@ -362,6 +362,17 @@ def generate_ism2_scene(
        else:
            y_meta = np.concatenate([y_meta, x_meta])

    # write individual ISM audio streams to the output file in an interleaved format
    audiofile.write(
        os.path.join(
            cfg.output_path,
            os.path.dirname(scene["name"]),
            cfg.use_output_prefix + os.path.basename(scene["name"]),
        ),
        y.audio,
        y.fs,
    )

    # write individual ISM metadata to output files in .csv format
    for i in range(N_sources):
        # generate .csv filename (should end with .0.csv, .1.csv, ...)