sonusai 0.16.0__py3-none-any.whl → 0.17.0__py3-none-any.whl

sonusai/summarize_metric_spenh.py

@@ -0,0 +1,71 @@
+"""sonusai summarize_metric_spenh
+
+usage: summarize_metric_spenh [-hr] [-s SORT] LOC
+
+options:
+    -h, --help
+    -s SORT, --sort SORT            Sort by SORT column. [default: MIXID]
+    -r, --reverse                   Reverse sort order.
+
+Summarize speech enhancement metrics results using data generated by SonusAI calc_metric_spenh.
+
+Inputs:
+    LOC         A SonusAI calc_metric_spenh results directory.
+
+"""
+import signal
+
+
+def signal_handler(_sig, _frame):
+    import sys
+
+    from sonusai import logger
+
+    logger.info('Canceled due to keyboard interrupt')
+    sys.exit(1)
+
+
+signal.signal(signal.SIGINT, signal_handler)
+
+
+def summarize_metric_spenh(location: str, by: str = 'MIXID', reverse: bool = False) -> str:
+    import glob
+
+    import pandas as pd
+
+    files = sorted(glob.glob(location + '/*_metric_spenh.txt'))
+    need_header = True
+    header = ['MIXID']
+    data = []
+    for file in files:
+        with open(file, 'r') as f:
+            for i, line in enumerate(f):
+                if i == 1 and need_header:
+                    need_header = False
+                    header.extend(line.strip().split())
+                elif i == 2:
+                    data.append(line.strip().split())
+                    break
+
+    df = pd.DataFrame(data, columns=header)
+    df[header[0:-2]] = df[header[0:-2]].apply(pd.to_numeric, errors='coerce')
+    return df.sort_values(by=by, ascending=not reverse).to_string(index=False)
+
+
+def main():
+    from docopt import docopt
+
+    import sonusai
+    from sonusai.utils import trim_docstring
+
+    args = docopt(trim_docstring(__doc__), version=sonusai.__version__, options_first=True)
+
+    by = args['--sort']
+    reverse = args['--reverse']
+    location = args['LOC']
+
+    print(summarize_metric_spenh(location, by, reverse))
+
+
+if __name__ == '__main__':
+    main()

sonusai/torchl_onnx-old.py

@@ -0,0 +1,216 @@
+"""sonusai torchl_onnx
+
+usage: torchl_onnx [-hv] [-b BATCH] [-t TSTEPS] [-o OUTPUT] MODEL CKPT
+
+options:
+    -h, --help
+    -v, --verbose                   Be verbose
+    -b BATCH, --batch BATCH         Batch size [default: 1]
+    -t TSTEPS, --tsteps TSTEPS      Timesteps [default: 1]
+    -o OUTPUT, --output OUTPUT      Output directory.
+
+Convert a trained Pytorch Lightning model to ONNX.  The model is specified as an
+sctl_*.py model file (sctl: sonusai custom torch lightning) and a checkpoint file
+for loading weights.
+
+Inputs:
+    MODEL       SonusAI Python custom model file.
+    CKPT        A Pytorch Lightning checkpoint file
+    BATCH       Batch size used in onnx conversion, overrides value in model ckpt. Defaults to 1.
+    TSTEPS      Timestep dimension size using in onnx conversion, overrides value in model ckpt if
+                the model has a timestep dimension.  Else it is ignored.
+
+Outputs:
+    OUTPUT/     A directory containing:
+                    <CKPT>.onnx        Model file with batch_size and timesteps equal to provided parameters
+                    <CKPT>-b1.onnx     Model file with batch_size=1 and if the timesteps dimension exists it
+                                       is set to 1 (useful for real-time inference applications)
+                    torchl_onnx.log
+
+Results are written into subdirectory <MODEL>-<TIMESTAMP> unless OUTPUT is specified.
+
+"""
+from sonusai import logger
+
+
+def main() -> None:
+    from docopt import docopt
+
+    import sonusai
+    from sonusai.utils import trim_docstring
+
+    args = docopt(trim_docstring(__doc__), version=sonusai.__version__, options_first=True)
+
+    verbose = args['--verbose']
+    batch_size = args['--batch']
+    timesteps = args['--tsteps']
+    model_path = args['MODEL']
+    ckpt_path = args['CKPT']
+    output_dir = args['--output']
+
+    from os import makedirs
+    from os.path import basename, splitext
+    from sonusai.utils import import_keras_model
+
+    # Import model definition file first to check
+    model_base = basename(model_path)
+    model_root = splitext(model_base)[0]
+    logger.info(f'Importing model from {model_base}')
+    try:
+        litemodule = import_keras_model(model_path)  # note works for pytorch lightning as well as keras
+    except Exception as e:
+        logger.exception(f'Error: could not import model from {model_path}: {e}')
+        raise SystemExit(1)
+
+    # Load checkpoint first to get hparams if available
+    from torch import load as load
+    ckpt_base = basename(ckpt_path)
+    ckpt_root = splitext(ckpt_base)[0]
+    logger.info(f'Loading checkpoint from {ckpt_base}')
+    try:
+        checkpoint = load(ckpt_path, map_location=lambda storage, loc: storage)
+    except Exception as e:
+        logger.exception(f'Error: could not load checkpoint from {ckpt_path}: {e}')
+        raise SystemExit(1)
+
+    from os.path import join, isdir, dirname, exists
+    from sonusai import create_file_handler
+    from sonusai import initial_log_messages
+    from sonusai import update_console_handler
+    from torch import randn
+    from sonusai.utils import create_ts_name
+
+    from sonusai.utils import create_ts_name
+    from torchinfo import summary
+
+    if batch_size is not None:
+        batch_size = int(batch_size)
+    if batch_size != 1:
+        batch_size = 1
+        logger.info(f'For now prediction only supports batch_size = 1, forcing it to 1 now')
+
+    if timesteps is not None:
+        timesteps = int(timesteps)
+
+    if output_dir is None:
+        output_dir = dirname(ckpt_path)
+    else:
+        if not isdir(output_dir):
+            makedirs(output_dir, exist_ok=True)
+
+    ofname = join(output_dir, ckpt_root + '.onnx')
+    # First try, then add date
+    if exists(ofname):
+        # add hour-min-sec if necessary
+        from datetime import datetime
+        ts = datetime.now()
+        ofname = join(output_dir, ckpt_root + '-' + ts.strftime('%Y%m%d') + '.onnx')
+    ofname_root = splitext(ofname)[0]
+
+    # Setup logging file
+    create_file_handler(ofname_root + '-onnx.log')
+    update_console_handler(verbose)
+    initial_log_messages('torchl_onnx')
+    logger.info(f'Imported model from {model_base}')
+    logger.info(f'Loaded checkpoint from {ckpt_base}')
+
+    if 'hyper_parameters' in checkpoint:
+        hparams = checkpoint['hyper_parameters']
+        logger.info(f'Found hyper-params on checkpoint named {checkpoint["hparams_name"]} '
+                    f'with {len(hparams)} total hparams.')
+        if batch_size is not None and hparams['batch_size'] != batch_size:
+            if batch_size != 1:
+                batch_size = 1
+                logger.info(f'For now prediction only supports batch_size = 1, forcing it to 1 now')
+            logger.info(f'Overriding batch_size: default = {hparams["batch_size"]}; specified = {batch_size}.')
+            hparams["batch_size"] = batch_size
+
+        if timesteps is not None:
+            if hparams['timesteps'] == 0 and timesteps != 0:
+                logger.warning(f'Model does not contain timesteps; ignoring override.')
+                timesteps = 0
+
+            if hparams['timesteps'] != 0 and timesteps == 0:
+                logger.warning(f'Model contains timesteps; ignoring override of 0, using model default.')
+                timesteps = hparams['timesteps']
+
+            if hparams['timesteps'] != timesteps:
+                logger.info(f'Overriding timesteps: default = {hparams["timesteps"]}; specified = {timesteps}.')
+                hparams['timesteps'] = timesteps
+
+        logger.info(f'Building model with hparams and batch_size={batch_size}, timesteps={timesteps}')
+        try:
+            model = litemodule.MyHyperModel(**hparams)  # use hparams
+            # litemodule.MyHyperModel.load_from_checkpoint(ckpt_name, **hparams)
+        except Exception as e:
+            logger.exception(f'Error: model build (MyHyperModel) in {model_base} failed: {e}')
+            raise SystemExit(1)
+    else:
+        logger.info(f'Warning: found checkpoint with no hyper-parameters, building model with defaults')
+        try:
+            tmp = litemodule.MyHyperModel()  # use default hparams
+        except Exception as e:
+            logger.exception(f'Error: model build (MyHyperModel) in {model_base} failed: {e}')
+            raise SystemExit(1)
+
+        if batch_size is not None:
+            if tmp.batch_size != batch_size:
+                logger.info(f'Overriding batch_size: default = {tmp.batch_size}; specified = {batch_size}.')
+        else:
+            batch_size = tmp.batch_size  # inherit
+
+        if timesteps is not None:
+            if tmp.timesteps == 0 and timesteps != 0:
+                logger.warning(f'Model does not contain timesteps; ignoring override.')
+                timesteps = 0
+
+            if tmp.timesteps != 0 and timesteps == 0:
+                logger.warning(f'Model contains timesteps; ignoring override.')
+                timesteps = tmp.timesteps
+
+            if tmp.timesteps != timesteps:
+                logger.info(f'Overriding timesteps: default = {tmp.timesteps}; specified = {timesteps}.')
+        else:
+            timesteps = tmp.timesteps
+
+        logger.info(f'Building model with default hparams and batch_size= {batch_size}, timesteps={timesteps}')
+        model = litemodule.MyHyperModel(timesteps=timesteps, batch_size=batch_size)
+
+    logger.info('')
+    # logger.info(summary(model))
+    # from lightning.pytorch import Trainer
+    # from lightning.pytorch.callbacks import ModelSummary
+    # trainer = Trainer(callbacks=[ModelSummary(max_depth=2)])
+    # logger.info(trainer.summarize())
+    logger.info('')
+    logger.info(f'feature       {model.hparams.feature}')
+    logger.info(f'num_classes   {model.num_classes}')
+    logger.info(f'batch_size    {model.hparams.batch_size}')
+    logger.info(f'timesteps     {model.hparams.timesteps}')
+    logger.info(f'flatten       {model.flatten}')
+    logger.info(f'add1ch        {model.add1ch}')
+    logger.info(f'truth_mutex   {model.truth_mutex}')
+    logger.info(f'input_shape   {model.input_shape}')
+    logger.info('')
+    logger.info(f'Loading weights from {ckpt_base}')
+    # model = model.load_from_checkpoint(ckpt_path)     # weights only, has problems - needs investigation
+    model.load_state_dict(checkpoint["state_dict"])
+    model.eval()
+    insample_shape = model.input_shape
+    insample_shape.insert(0, batch_size)
+    input_sample = randn(insample_shape)
+    logger.info(f'Creating onnx model ...')
+    for m in model.modules():
+        if 'instancenorm' in m.__class__.__name__.lower():
+            logger.info(f'Forcing train=false for instancenorm instance {m}, {m.__class__.__name__.lower()}')
+            m.train(False)
+            # m.track_running_stats=True  # has problems
+    model.to_onnx(file_path=ofname, input_sample=input_sample, export_params=True)
+
+
+if __name__ == '__main__':
+    try:
+        main()
+    except KeyboardInterrupt:
+        logger.info('Canceled due to keyboard interrupt')
+        exit()

sonusai/tplot.py

@@ -41,7 +41,19 @@ options:
    A multi-page plot TARGET-tplot.pdf or CONFIG-tplot.pdf is generated.
 
 """
-from sonusai import logger
+import signal
+
+
+def signal_handler(_sig, _frame):
+    import sys
+
+    from sonusai import logger
+
+    logger.info('Canceled due to keyboard interrupt')
+    sys.exit(1)
+
+
+signal.signal(signal.SIGINT, signal_handler)
 
 
 # TODO: re-work for modern mixdb API
@@ -328,8 +340,4 @@ def main() -> None:
 
 
 if __name__ == '__main__':
-    try:
-        main()
-    except KeyboardInterrupt:
-        logger.info('Canceled due to keyboard interrupt')
-        raise SystemExit(0)
+    main()

sonusai/utils/onnx_utils.py

@@ -1,6 +1,12 @@
 from dataclasses import dataclass
 
+from sonusai import logger
+from typing import Any  #List, Optional, Tuple
+import onnxruntime as ort
 from onnxruntime import InferenceSession
+import onnx
+from onnx import ValueInfoProto
+from os.path import basename, splitext, exists, isfile
 
 
 @dataclass(frozen=True)
@@ -14,52 +20,135 @@ class SonusAIMetaData:
     feature: str
 
 
-def add_sonusai_metadata(model,
-                         is_flattened: bool = True,
-                         has_timestep: bool = True,
-                         has_channel: bool = False,
-                         is_mutex: bool = True,
-                         feature: str = ''):
-    """Add SonusAI metadata to an ONNX model.
+def get_and_check_inputs(model: onnx.ModelProto) -> tuple[list[ValueInfoProto], list[list[int] | str]]:
+    # ignore initializer inputs (only seen in older onnx < v1.5
+    initializer_names = [x.name for x in model.graph.initializer]
+    onnx_inputs = [ipt for ipt in model.graph.input if ipt.name not in initializer_names]
+    if len(onnx_inputs) != 1:
+        logger.warning(f'Warning: onnx model does not have 1 input, but {len(onnx_inputs)}')
+        #raise SystemExit(1)
+
+    inshapes = []
+    for inp in onnx_inputs:                      # iterate through inputs of the graph to find shapes
+        tensor_type = inp.type.tensor_type       # get tensor type: 0, 1, 2,
+        if (tensor_type.HasField("shape")):      # check if it has a shape:
+            tmpshape = []
+            for d in tensor_type.shape.dim:      # iterate through dimensions of the shape
+                if (d.HasField("dim_value")):    # known dimension, int value
+                    tmpshape.append(d.dim_value)
+                elif (d.HasField("dim_param")):  # dynamic dim with symbolic name of d.dim_param
+                    tmpshape.append(0)           # set size to 0
+                else:                            # unknown dimension with no name
+                    tmpshape.append(0)           # also set to 0
+            inshapes.append(tmpshape)            # add as a list
+        else:
+            inshapes.append("unknown rank")
+
+    # This one-liner works only if input has type and shape, returns a list
+    #in0shape = [d.dim_value for d in onnx_inputs[0].type.tensor_type.shape.dim]
+
+    return onnx_inputs, inshapes
 
-    :param model: ONNX model
-    :param is_flattened: Model feature data is flattened
-    :param has_timestep: Model has timestep dimension
-    :param has_channel: Model has channel dimension
-    :param is_mutex: Model label output is mutually exclusive
-    :param feature: Model feature type
-    """
-    is_flattened_flag = model.metadata_props.add()
-    is_flattened_flag.key = 'is_flattened'
-    is_flattened_flag.value = str(is_flattened)
 
-    has_timestep_flag = model.metadata_props.add()
-    has_timestep_flag.key = 'has_timestep'
-    has_timestep_flag.value = str(has_timestep)
+def get_and_check_outputs(model: onnx.ModelProto) -> tuple[list[ValueInfoProto], list[list[int | Any] | str]]:
+    onnx_outputs = [opt for opt in model.graph.output]
+    if len(onnx_outputs) != 1:
+        logger.warning(f'Warning: onnx model does not have 1 output, but {len(onnx_outputs)}')
 
-    has_channel_flag = model.metadata_props.add()
-    has_channel_flag.key = 'has_channel'
-    has_channel_flag.value = str(has_channel)
+    oshapes = []
+    for inp in onnx_outputs:                     # iterate through inputs of the graph to find shapes
+        tensor_type = inp.type.tensor_type       # get tensor type: 0, 1, 2,
+        if (tensor_type.HasField("shape")):      # check if it has a shape:
+            tmpshape = []
+            for d in tensor_type.shape.dim:      # iterate through dimensions of the shape
+                if (d.HasField("dim_value")):    # known dimension, int value
+                    tmpshape.append(d.dim_value)
+                elif (d.HasField("dim_param")):  # dynamic dim with symbolic name of d.dim_param
+                    tmpshape.append(0)           # set size to 0
+                else:                            # unknown dimension with no name
+                    tmpshape.append(0)           # also set to 0
+            oshapes.append(tmpshape)             # add as a list
+        else:
+            oshapes.append("unknown rank")
 
-    is_mutex_flag = model.metadata_props.add()
-    is_mutex_flag.key = 'is_mutex'
-    is_mutex_flag.value = str(is_mutex)
+    return onnx_outputs, oshapes
 
-    feature_flag = model.metadata_props.add()
-    feature_flag.key = 'feature'
-    feature_flag.value = str(feature)
+
+def add_sonusai_metadata(model, hparams):
+    """Add SonusAI hyper-parameter metadata to an ONNX model using key hparams
+
+    :param model: ONNX model
+    :hparams: dictionary of hyper-parameters, added
+    Note SonusAI conventions require models to have:
+      - feature: Model feature type
+      - is_flattened: Model input feature data is flattened (stride + bins combined)
+      - timesteps: Size of timestep dimension (0 for no dimension)
+      - add1ch:   Model input has channel dimension
+      - truth_mutex: Model label output is mutually exclusive
+    """
+
+    # Add hyper-parameters as metadata in onnx model under hparams key
+    assert eval(str(hparams)) == hparams       # Note hparams should be a dict (i.e. extracted from checkpoint)
+    meta = model.metadata_props.add()
+    meta.key = "hparams"
+    meta.value = str(hparams)
 
     return model
 
 
-def get_sonusai_metadata(model: InferenceSession) -> SonusAIMetaData:
-    """Get SonusAI metadata from an ONNX model.
+def get_sonusai_metadata(session: InferenceSession) -> SonusAIMetaData:
+    """Get SonusAI hyper-parameter metadata from an ONNX Runtime session.
+       Returns dictionary hparams
     """
-    m = model.get_modelmeta().custom_metadata_map
-    return SonusAIMetaData(input_shape=model.get_inputs()[0].shape,
-                           output_shape=model.get_outputs()[0].shape,
-                           flattened=m['is_flattened'] == 'True',
-                           timestep=m['has_timestep'] == 'True',
-                           channel=m['has_channel'] == 'True',
-                           mutex=m['is_mutex'] == 'True',
-                           feature=m['feature'])
+    meta = session.get_modelmeta()
+    hparams = eval(meta.custom_metadata_map["hparams"])
+
+    # m = model.get_modelmeta().custom_metadata_map
+    # return SonusAIMetaData(input_shape=model.get_inputs()[0].shape,
+    #                        output_shape=model.get_outputs()[0].shape,
+    #                        flattened=m['is_flattened'] == 'True',
+    #                        timestep=m['has_timestep'] == 'True',
+    #                        channel=m['has_channel'] == 'True',
+    #                        mutex=m['is_mutex'] == 'True',
+    #                        feature=m['feature'])
+
+    return hparams
+
+def load_ort_session(model_path, providers=['CPUExecutionProvider']):
+    if exists(model_path) and isfile(model_path):
+        model_basename = basename(model_path)
+        model_root = splitext(model_basename)[0]
+        logger.info(f'Importing model from {model_basename}')
+        try:
+            session = ort.InferenceSession(model_path, providers=providers)
+            options = ort.SessionOptions()
+        except Exception as e:
+            logger.exception(f'Error: could not load onnx model from {model_path}: {e}')
+            raise SystemExit(1)
+    else:
+        logger.exception(f'Error: model file does not exist: {model_path}')
+        raise SystemExit(1)
+
+    logger.info(f'Opened session with provider options: {session._provider_options}.')
+    try:
+        meta = session.get_modelmeta()
+        hparams = eval(meta.custom_metadata_map["hparams"])
+        logger.info(f'Sonusai hyper-parameter metadata was found in model with {len(hparams)} parameters, '
+                    f'checking for required ones ...')
+        # Print to log here will fail if required parameters not available.
+        logger.info(f'feature         {hparams["feature"]}')
+        logger.info(f'batch_size      {hparams["batch_size"]}')
+        logger.info(f'timesteps       {hparams["timesteps"]}')
+        logger.info(f'flatten, add1ch {hparams["flatten"]}, {hparams["add1ch"]}')
+        logger.info(f'truth_mutex     {hparams["truth_mutex"]}')
+    except:
+        hparams = None
+        logger.warning(f'Warning: onnx model does not have required SonusAI hyper-parameters.')
+
+    inputs = session.get_inputs()
+    outputs = session.get_outputs()
+
+    #in_names = [n.name for n in session.get_inputs()]
+    #out_names = [n.name for n in session.get_outputs()]
+
+    return session, options, model_root, hparams, inputs, outputs

{sonusai-0.16.0.dist-info → sonusai-0.17.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sonusai
-Version: 0.16.0
+Version: 0.17.0
 Summary: Framework for building deep neural network models for sound, speech, and voice AI
 Home-page: https://aaware.com
 License: GPL-3.0-only

{sonusai-0.16.0.dist-info → sonusai-0.17.0.dist-info}/RECORD

@@ -1,7 +1,8 @@
-sonusai/__init__.py,sha256=ziSNMMCNS-tIp2ON1e7ST_xNLT-P8BvJfV2IxUvSJZs,2896
+sonusai/__init__.py,sha256=vzTFfRB-NeO-Sm3puySDJOybk3ND_Oj6w0EejQPmH1U,2978
 sonusai/aawscd_probwrite.py,sha256=GukR5owp_0A3DrqSl9fHWULYgclNft4D5OkHIwfxxkc,3698
-sonusai/audiofe.py,sha256=v-GWrma5NNSukkPH427_6ilvFghUlDy4N9sfp_gvwCY,9303
-sonusai/calc_metric_spenh.py,sha256=zJD9cuO4iXvOTuqDjpv-tJQg9bC7r5L7K74Av16lOk4,61762
+sonusai/audiofe.py,sha256=yPtbxeRAzlnPcRESXKNVexvIm6fM4WVxEHjd0w9n5O0,12455
+sonusai/calc_metric_spenh-save.py,sha256=-LR5BtAnYNYKav1B2ZsB7gGevidCsQ91yFEaH8Ycyr8,61765
+sonusai/calc_metric_spenh.py,sha256=_92RWCyxAf7_S61L5oX6o4GuvuOCDKn7BPg7pB0r1kY,61836
 sonusai/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sonusai/data/genmixdb.yml,sha256=-XSs_hUR6wHJVoTPmSewzXL7u61X-xmHY46lNPatxSE,1025
 sonusai/data/speech_ma01_01.wav,sha256=PK0vMKg-NR6rPE3KouxHGF6PKXnJCr7AwjMqfu98LUA,76644
@@ -9,11 +10,11 @@ sonusai/data/whitenoise.wav,sha256=I2umov0m34y56F9IsIBi1XtE76ZeZaSKDf70cJRe3pI,1
 sonusai/doc/__init__.py,sha256=rP5Hgn0Iys_xkuv4caxngdqehuU4zLZsiKuv8Nde67M,19
 sonusai/doc/doc.py,sha256=3z210v6ZckuOlsGZ3ySQBdlCNmBp2M1ahqhqG_eUN58,22664
 sonusai/doc.py,sha256=l8CaFgLI8mqx4tn0aXfxKqa2dy9GgC0zjYxZAkpmi1E,878
-sonusai/genft.py,sha256=6hOds8d-pYRawesLYh7XLrQh4PweWUj8U5Cbzx45bNQ,5572
-sonusai/genmix.py,sha256=0AiUfF7n0CGOp5v-woNfeP3-QaVQUb0CJZc0oXkvqpk,7016
-sonusai/genmixdb.py,sha256=b-c22X7dd1ePLGrVuhVcn-AXHnxSrC5lTaR6KlrD1-4,19610
-sonusai/gentcst.py,sha256=8jeXirgJe0OGgknC8A-rIudjHeH8UTYPpuh71Ha-I3w,20165
-sonusai/lsdb.py,sha256=TTMQ-0H8fFzUSczt6yjy-9xUjZSdIGQzTVH5Xr6XPSA,5941
+sonusai/genft.py,sha256=OzET3iTE-QhrUckzidfZvCDXZlAxIF5Xe5NEf856Vvk,5662
+sonusai/genmix.py,sha256=TU5aTebGHsbfwsRbynYbegGBelSma9khuQkDk0dFE3I,7075
+sonusai/genmixdb.py,sha256=M67Y_SEysgHfTmHHOdOjxdpuryTMDNgbDteCzR1uLk8,19669
+sonusai/gentcst.py,sha256=W1ZO3xs7CoZkFcvOTH-FLJOIA4I7Wzb0HVRC3hGGSaM,20223
+sonusai/lsdb.py,sha256=fMRqPlAu4B-4MsTXX-NaWXYyJ_dAOJlS-LrvQPQQsXg,6028
 sonusai/main.py,sha256=GC-pQrSqx9tWwIcmEo6V9SraEv5KskBLS_W_wz-f2ZM,2509
 sonusai/metrics/__init__.py,sha256=56itZW3S1I7ZYvbxPmFIVPAh1AIJZdljByz1uCrHqFE,635
 sonusai/metrics/calc_class_weights.py,sha256=dyY7daEIf5Ms5tfTf6wF0fkx_GnMADHOZR_rtsfGoVM,3933
@@ -60,14 +61,20 @@ sonusai/mixture/truth_functions/file.py,sha256=jOJuC_3y9BH6GGOp9eKcbVrHLVRzUA80B
 sonusai/mixture/truth_functions/phoneme.py,sha256=stYdlPuNytQK_LLT61OJLfYSqKd-sDjQZdtJKGzt5wA,479
 sonusai/mixture/truth_functions/sed.py,sha256=8cHjEFjZaH_0hIOHhPmj4AJz2GpEADM6Ys2x4NoiWSY,2469
 sonusai/mixture/truth_functions/target.py,sha256=KAsjugDRooOA5BRcHVAbZRgV7l8S5CFg7CZ0XtKZaQ0,5764
-sonusai/mkmanifest.py,sha256=r6bQ6uoBVJIAviqNtJo7AZjXS8EhxdSAW-NHlhPzDCU,8646
-sonusai/mkwav.py,sha256=LZNyhq4gJEs_NtGvRsYHA2qfgkkODpt6HoH1b-Tjjuw,5266
-sonusai/onnx_predict.py,sha256=Bz_pR28oAZBarNajlKwyzBxmW7ktum77SmxDN2onKPM,9060
-sonusai/plot.py,sha256=u-PvF8guNcm0b-GN99xfEkrcAAtidAEY3RLDzNvcyYk,17014
-sonusai/post_spenh_targetf.py,sha256=NIMhDXeDuUqeWukNaAUMvDw9JpEVCauwjrL2F4M9nrI,4927
+sonusai/mkmanifest.py,sha256=7lfK7YOdgAEP_Lxrf-YDxZ5iLH9MJuaOltBVpav2M9M,8705
+sonusai/mkwav.py,sha256=kLfC2ZuF-t8P97nqYw2falTZpymxAeXv0YTJCe6nK10,5356
+sonusai/onnx_predict-old.py,sha256=Bz_pR28oAZBarNajlKwyzBxmW7ktum77SmxDN2onKPM,9060
+sonusai/onnx_predict-save.py,sha256=ewiV5-HcW5zcDWuIF9xEbdBwbdL8vNWfu2_kaur5jAo,22354
+sonusai/onnx_predict.py,sha256=-ETvGH7fHXjnmY-c2r_8gOHEX1VSLwdSToZ6gSBP3_w,23021
+sonusai/ovino_predict.py,sha256=QtWY_YEdaeqJL5yikfeeDwhbBtjXYjGNtR9310PGGSc,21830
+sonusai/ovino_query_devices.py,sha256=XkXdOlZldI0MfG6nXZfTq8OgECaY8gAzQl36sJRuaIU,1584
+sonusai/plot.py,sha256=ERkmxMM3qjcCDm4LGDQY4fRAncCYAzP7uW8iZ7_brcg,17105
+sonusai/post_spenh_targetf.py,sha256=xOz5T6WZuyTHmfbtILIY9skgH064Wvi2GF2Bo5L3YMU,4998
 sonusai/queries/__init__.py,sha256=oKY5JeqZ4Cz7DwCwPc1_ydB8bUs6KaMcWFp_w02TjOs,255
 sonusai/queries/queries.py,sha256=FNMUKnoY_Ya9S5sNhsB8ppwy0B7V55ilbbjhQRv_UN8,7552
-sonusai/tplot.py,sha256=yFyyyg9ymp2Eh-64Muu0EFFEY61MoJSV0a_fy9OWaCk,14485
+sonusai/summarize_metric_spenh.py,sha256=OiZe_bhCq5esXNhsOkHDD7g4ssYrpENDHvDVoPzV9iw,1822
+sonusai/torchl_onnx-old.py,sha256=5JYow3XpBaUdtuyAW0mOZyCKL_4FrHvEekYBRdDT6KA,8967
+sonusai/tplot.py,sha256=85T6OPZfxVegHBiSuilFpdgCNMEE0VKAuciNy4rCY5Y,14544
 sonusai/utils/__init__.py,sha256=TCXlcW8W0Up2f5ciSgz3DabvH1MxrrWD0LK6pQTJkeA,2215
 sonusai/utils/asl_p56.py,sha256=-bvQpd-jRQVURbkZJpRoyEAq6gTv9Rc3oFDbh5_lcjY,3861
 sonusai/utils/asr.py,sha256=6y6VYJizHpuQ3MgKbEQ4t2gofO-MW6Ez23oAd6d23IE,2920
@@ -96,7 +103,7 @@ sonusai/utils/human_readable_size.py,sha256=SjYT0fUlpbfCzCXHo6csir-VMwqfs5ogr-fg
 sonusai/utils/max_text_width.py,sha256=pxiJMwb_zlkNntexgo7S6lAuF7NLLZvFdOCkxdsQJVY,315
 sonusai/utils/model_utils.py,sha256=lt2KOGJqsinG71W0i3U29UXFO-47GMAlEabsf2um7bA,862
 sonusai/utils/numeric_conversion.py,sha256=GRO_2Fba8CcxcFY7bEXKOEUEUX6neA-VN__Bxi1ULsE,340
-sonusai/utils/onnx_utils.py,sha256=BRsHGlcu5L0v_1z83MNy8TAcBeb7tJd_4OBJgOMLen8,2200
+sonusai/utils/onnx_utils.py,sha256=L0BcwF0or1UwxYWzvWNWNJHKvG_oEmI2AxYh4msp2vc,6862
 sonusai/utils/parallel.py,sha256=bxedjCzBv9oxzU7NajRr6mOKmkCWr2P7FWAI0p2p9N8,1981
 sonusai/utils/print_mixture_details.py,sha256=BzYM4-wHHNa6zxPzBMUJxwKt0gKHmvbwdd7Yp0w15Yk,3017
 sonusai/utils/ranges.py,sha256=NPBZOVzMb95GTOIxltVO-wSzgcXqZ14wbdV46JDLKrw,1222
@@ -109,7 +116,7 @@ sonusai/utils/stratified_shuffle_split.py,sha256=rJNXvBp-GxoKzH3OpL7k0ANSu5xMP2z
 sonusai/utils/wave.py,sha256=O4ZXkZ6wjrKGa99wBCdFd8G6bp91MXXDnmGihpaEMh0,856
 sonusai/utils/yes_or_no.py,sha256=eMLXBVH0cEahiXY4W2KNORmwNQ-ba10eRtldh0y4NYg,263
 sonusai/vars.py,sha256=m2AefF0m5bXWGXpJj8Pi42zWL2ydeEj7bkak3GrtMyM,940
-sonusai-0.16.0.dist-info/METADATA,sha256=MWrFI-GR33t2hVwlEwtLzzfEZiqO3mekRf7P7L8TaSk,2443
-sonusai-0.16.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-sonusai-0.16.0.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
-sonusai-0.16.0.dist-info/RECORD,,
+sonusai-0.17.0.dist-info/METADATA,sha256=wn6MaT5JXlVzp45_huLIAnJeH04kQZm8r0_vWiDX3LU,2443
+sonusai-0.17.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+sonusai-0.17.0.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
+sonusai-0.17.0.dist-info/RECORD,,

{sonusai-0.16.0.dist-info → sonusai-0.17.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 1.8.1
 Root-Is-Purelib: true
 Tag: py3-none-any