Commit c74b0e75 authored by Jan Kiene's avatar Jan Kiene
Browse files

add osba split markers

parent 2d481a89

tests/conftest.py

+5 −0
Original line number Diff line number Diff line
@@ -1265,6 +1265,11 @@ def get_split_idx(input_file: str, sampling_rate_khz: int) -> Optional[np.ndarra 
    input_file = input_file.lower()

    if "omasa" in input_file:

        format = "_".join(input_file.replace(".wav", "").split("_")[:-1])

    elif "osba" in input_file:

        if "foa" in input_file:

            format = "osba_foa"

        else:

            format = "osba_hoa"

    else:

        format = input_file.replace(".wav", "").split("_")[-1].lower()

    idx = SPLIT_IDX.get(format, np.empty(0))

tests/constants.py

+54 −0
Original line number Diff line number Diff line
@@ -379,6 +379,58 @@ SPLIT_IDX_LTV_OMASA_4ISM_16KHZ = np.asarray( 
        1878304,

    ]

)

SPLIT_IDX_LTV_OSBA_FOA_16KHZ = np.asarray(

    [

        110243,

        176353,

        231257,

        327228,

        406430,

        453959,

        465000,

        496215,

        565603,

        735726,

        846637,

        1166658,

        1316644,

        1391826,

        1697613,

        1716813,

        1812764,

        1908681,

        2036312,

        2193583,

        2361556,

    ]

)

SPLIT_IDX_LTV_OSBA_HOA_16KHZ = np.asarray(

    [

        69324,

        169120,

        339217,

        450133,

        739726,

        930163,

        1090182,

        1240385,

        1316803,

        1617775,

        1640120,

        1704436,

        1759348,

        1854498,

        1887201,

        1932676,

        1983352,

        1991890,

        2024762,

        2136276,

        2231825,

        2328364,

        2424105,

    ]

)



SPLIT_IDX = {

    "stereo": SPLIT_IDX_LTV_STEREO_16KHZ,
@@ -402,4 +454,6 @@ SPLIT_IDX = { 
    "omasa_2ism": SPLIT_IDX_LTV_OMASA_2ISM_16KHZ,

    "omasa_3ism": SPLIT_IDX_LTV_OMASA_3ISM_16KHZ,

    "omasa_4ism": SPLIT_IDX_LTV_OMASA_4ISM_16KHZ,

    "osba_foa": SPLIT_IDX_LTV_OSBA_FOA_16KHZ,

    "osba_hoa": SPLIT_IDX_LTV_OSBA_HOA_16KHZ,

}