Commit 369dba93 authored by Vinit Veera's avatar Vinit Veera
Browse files

Updating the metadata paths and copying the metadata to the new folder.

parent 35e1cc00

ivas_processing_scripts/__init__.py

+4 −4
Original line number Diff line number Diff line
@@ -97,9 +97,6 @@ def main(args): 
        # set up logging

        logger = logging_init(args, cfg)



        # checking if audio is a multiple of frame size

        multiple_of_frame_size(cfg, logger)



        # Re-ordering items based on concatenation order

        if hasattr(cfg, "preprocessing_2"):

            if (
@@ -132,6 +129,9 @@ def main(args): 


        cfg.metadata_path = metadata



        # checking if audio is a multiple of frame size

        multiple_of_frame_size(cfg, logger)



        # run preprocessing only once

        if hasattr(cfg, "preprocessing"):

            # save process info for background noise
@@ -181,7 +181,7 @@ def main(args): 
        if cfg.condition_in_output_filename:

            rename_generated_conditions(cfg.output_path)



        compare_wav_lengths(cfg.input_path, cfg.output_path, logger)

        # compare_wav_lengths(cfg.input_path, cfg.output_path, logger)



    # copy configuration to output directory

    with open(cfg.output_path.joinpath(f"{cfg.name}.yml"), "w") as f:

ivas_processing_scripts/processing/processing.py

+8 −0
Original line number Diff line number Diff line
@@ -595,6 +595,14 @@ def multiple_of_frame_size( 
        # Update audio file path in list

        cfg.items_list[i] = output_dir / item.name



        # Copy metadata and update path

        if input_format.startswith("ISM") or input_format.startswith("MASA"):

            for j in range(int(cfg.input["fmt"][3])):

                copyfile(

                    cfg.metadata_path[i][j], output_dir / cfg.metadata_path[i][j].name

                )

                cfg.metadata_path[i][j] = output_dir / cfg.metadata_path[i][j].name



    # Check if all files are present in output directory

    all_files_present = all(

        [(output_dir / item.name).exists() for audio_file in cfg.items_list]