| | |
|---|
| | | import argparse |
|---|
| | | import logging |
|---|
| | | import os |
|---|
| | | import sys |
|---|
| | | import json |
|---|
| | | from pathlib import Path |
|---|
| | |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | | ncpu = kwargs.get("ncpu", 1) |
|---|
| | | torch.set_num_threads(ncpu) |
|---|
| | | |
|---|
| | | if batch_size > 1: |
|---|
| | | raise NotImplementedError("batch decoding is not implemented") |
|---|
| | | if ngpu > 1: |
|---|
| | |
|---|
| | | # do vad segment |
|---|
| | | _, results = speech2vadsegment(**batch) |
|---|
| | | for i, _ in enumerate(keys): |
|---|
| | | results[i] = json.dumps(results[i]) |
|---|
| | | if "MODELSCOPE_ENVIRONMENT" in os.environ and os.environ["MODELSCOPE_ENVIRONMENT"] == "eas": |
|---|
| | | results[i] = json.dumps(results[i]) |
|---|
| | | item = {'key': keys[i], 'value': results[i]} |
|---|
| | | vad_results.append(item) |
|---|
| | | if writer is not None: |
|---|
