wehub-resource-sync
ba4be087d5
Create PR to main with cherry-pick from release / cherry-pick (push) Failing after 0s
CICD NeMo / pre-flight (push) Failing after 0s
CICD NeMo / configure (push) Has been skipped
Build, validate, and release Neural Modules / pre-flight (push) Failing after 1s
CICD NeMo / code-linting (push) Has been skipped
Build, validate, and release Neural Modules / release (push) Has been skipped
Build, validate, and release Neural Modules / release-summary (push) Has been cancelled
CICD NeMo / cicd-test-container-build (push) Has been cancelled
CICD NeMo / cicd-import-tests (push) Has been cancelled
CICD NeMo / L0_Setup_Test_Data_And_Models (push) Has been cancelled
CICD NeMo / cicd-main-unit-tests (push) Has been cancelled
CICD NeMo / cicd-main-speech (push) Has been cancelled
CICD NeMo / Nemo_CICD_Test (push) Has been cancelled
CICD NeMo / Coverage (e2e) (push) Has been cancelled
CICD NeMo / Coverage (unit-test) (push) Has been cancelled
CodeQL / Analyze (python) (push) Has been cancelled
CICD NeMo / cicd-wait-in-queue (push) Has been cancelled
70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
# Copyright (c) 2022, NVIDIA CORPORATION. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import argparse
|
|
import random
|
|
|
|
from nemo.collections.asr.parts.utils.manifest_utils import create_manifest
|
|
|
|
random.seed(42)
|
|
|
|
"""
|
|
This script creates manifest file for speaker diarization inference purposes.
|
|
Useful to get manifest when you have list of audio files and optionally rttm and uem files for evaluation
|
|
|
|
Note: make sure basename for each file is unique and rttm files also has the corresponding base name for mapping
|
|
"""
|
|
|
|
|
|
def main(
|
|
wav_path, text_path=None, rttm_path=None, uem_path=None, ctm_path=None, manifest_filepath=None, add_duration=False
|
|
):
|
|
create_manifest(
|
|
wav_path,
|
|
manifest_filepath,
|
|
text_path=text_path,
|
|
rttm_path=rttm_path,
|
|
uem_path=uem_path,
|
|
ctm_path=ctm_path,
|
|
add_duration=add_duration,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--paths2audio_files", help="path to text file containing list of audio files", type=str, required=True
|
|
)
|
|
parser.add_argument("--paths2txt_files", help="path to text file containing list of transcription files", type=str)
|
|
parser.add_argument("--paths2rttm_files", help="path to text file containing list of rttm files", type=str)
|
|
parser.add_argument("--paths2uem_files", help="path to uem files", type=str)
|
|
parser.add_argument("--paths2ctm_files", help="path to ctm files", type=str)
|
|
parser.add_argument("--manifest_filepath", help="path to output manifest file", type=str, required=True)
|
|
parser.add_argument(
|
|
"--add_duration",
|
|
help="add duration of audio files to output manifest files.",
|
|
action='store_true',
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
main(
|
|
args.paths2audio_files,
|
|
args.paths2txt_files,
|
|
args.paths2rttm_files,
|
|
args.paths2uem_files,
|
|
args.paths2ctm_files,
|
|
args.manifest_filepath,
|
|
args.add_duration,
|
|
)
|