sonusai 0.18.7__py3-none-any.whl → 0.18.8__py3-none-any.whl

sonusai/__init__.py

@@ -24,7 +24,6 @@ commands_doc = """
    mkwav                        Make WAV files from a mixture database
    onnx_predict                 Run ONNX predict on a trained model
    plot                         Plot mixture data
-   post_spenh_targetf           Run post-processing for speech enhancement targetf data
    summarize_metric_spenh       Summarize speech enhancement and analysis results
    tplot                        Plot truth data
    vars                         List custom SonusAI variables

sonusai/mixture/helpers.py

@@ -276,7 +276,6 @@ def read_mixture_data(name: str, items: list[str] | str) -> Any:
     :return: Data (or tuple of data)
     """
     from os.path import exists
-    from typing import Any
 
     import h5py
     import numpy as np
@@ -287,7 +286,10 @@ def read_mixture_data(name: str, items: list[str] | str) -> Any:
         if d_name in file:
             data = np.array(file[d_name])
             if data.size == 1:
-                return data.item()
+                item = data.item()
+                if isinstance(item, bytes):
+                    return item.decode('utf-8')
+                return item
             return data
         return None
 

sonusai/mixture/mixdb.py

@@ -214,6 +214,10 @@ class MixtureDatabase:
             MetricDoc('Truth Metrics', 'sedtopn', '(not implemented) N most active by largest sedavg [N, 1]'),
         ])
         for name in self.asr_configs:
+            metrics.append(MetricDoc('Target Metrics', f'tasr.{name}',
+                                     f'Target[0] ASR text using {name} ASR as defined in mixdb asr_configs parameter'))
+            metrics.append(MetricDoc('Mixture Metrics', f'mxasr.{name}',
+                                     f'ASR text using {name} ASR as defined in mixdb asr_configs parameter'))
             metrics.append(MetricDoc('Mixture Metrics', f'mxwer.{name}',
                                      f'Word error rate using {name} ASR as defined in mixdb asr_configs parameter'))
 
@@ -1185,7 +1189,7 @@ class MixtureDatabase:
 
     def mixture_metrics(self, m_id: int,
                         metrics: list[str],
-                        force: bool = False) -> list[float | int | Segsnr]:
+                        force: bool = False) -> list[float | int | str | Segsnr]:
         """Get metrics data for the given mixture ID
 
         :param m_id: Zero-based mixture ID
@@ -1328,7 +1332,56 @@ class MixtureDatabase:
 
         noise_stats = create_noise_stats()
 
-        def calc(m: str) -> float | int | Segsnr:
+        def create_asr_config() -> Callable[[str], dict]:
+            state: dict[str, dict] = {}
+
+            def get(asr_name) -> dict:
+                nonlocal state
+                if asr_name not in state:
+                    state[asr_name] = self.asr_configs.get(asr_name, None)
+                    if state[asr_name] is None:
+                        raise SonusAIError(f"Unrecognized ASR name: '{asr_name}'")
+                return state[asr_name]
+
+            return get
+
+        asr_config = create_asr_config()
+
+        def create_target_asr() -> Callable[[str], str]:
+            state: dict[str, str] = {}
+
+            def get(asr_name) -> str:
+                nonlocal state
+                if asr_name not in state:
+                    state[asr_name] = calc_asr(target_audio(), **asr_config(asr_name)).text
+                return state[asr_name]
+
+            return get
+
+        target_asr = create_target_asr()
+
+        def create_mixture_asr() -> Callable[[str], str]:
+            state: dict[str, str] = {}
+
+            def get(asr_name) -> str:
+                nonlocal state
+                if asr_name not in state:
+                    state[asr_name] = calc_asr(mixture_audio(), **asr_config(asr_name)).text
+                return state[asr_name]
+
+            return get
+
+        mixture_asr = create_mixture_asr()
+
+        def get_asr_name(m: str) -> str:
+            parts = m.split('.')
+            if len(parts) != 2:
+                raise SonusAIError(
+                    f"Unrecognized format: '{m}'; must be of the form: '<metric>.<name>'")
+            asr_name = parts[1]
+            return asr_name
+
+        def calc(m: str) -> float | int | str | Segsnr:
             if m == 'mxsnr':
                 return self.mixture(m_id).snr
 
@@ -1340,31 +1393,21 @@ class MixtureDatabase:
 
             # Otherwise, generate data as needed
             if m.startswith('mxwer'):
-                parts = m.split('.')
-                if len(parts) != 2:
-                    raise SonusAIError(
-                        f"Unrecognized 'mxwer' format: '{m}'; must be of the form: 'mxwer.<name>'")
-                asr_name = parts[1]
-                asr_config = self.asr_configs.get(asr_name, None)
-                if asr_config is None:
-                    raise SonusAIError(f"Unrecognized metric: '{m}'")
+                asr_name = get_asr_name(m)
 
                 if self.mixture(m_id).snr < -96:
                     # noise only, ignore/reset target asr
                     return float('nan')
 
-                # ignore mixup
-                target_asr = self.mixture_speech_metadata(m_id, 'text')[0]
-                if target_asr is None:
-                    target_asr = calc_asr(target_audio(), **asr_config).text
-
-                if target_asr:
-                    mixture_asr = calc_asr(mixture_audio(), **asr_config).text
-                    return calc_wer(mixture_asr, target_asr).wer * 100
+                if target_asr(asr_name):
+                    return calc_wer(mixture_asr(asr_name), target_asr(asr_name)).wer * 100
 
                 # TODO: should this be NaN like above?
                 return float(0)
 
+            if m.startswith('mxasr'):
+                return mixture_asr(get_asr_name(m))
+
             if m == 'mxssnr_avg':
                 return calc_segsnr_f(segsnr_f()).avg
 
@@ -1454,6 +1497,9 @@ class MixtureDatabase:
             if m == 'tpkc':
                 return target_stats().pkc
 
+            if m.startswith('tasr'):
+                return target_asr(get_asr_name(m))
+
             if m == 'ndco':
                 return noise_stats().dco
 
@@ -1501,7 +1547,7 @@ class MixtureDatabase:
 
             raise SonusAIError(f"Unrecognized metric: '{m}'")
 
-        result: list[float | int | Segsnr] = []
+        result: list[float | int | str | Segsnr] = []
         for metric in metrics:
             result.append(calc(metric))
 

{sonusai-0.18.7.dist-info → sonusai-0.18.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sonusai
-Version: 0.18.7
+Version: 0.18.8
 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.18.7.dist-info → sonusai-0.18.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-sonusai/__init__.py,sha256=zV3NEEnPJs_dFdCpCS5dM8vJ8nW1bvNQDSNVbJOU7KQ,3150
+sonusai/__init__.py,sha256=PakKXwYWgB0TZysZ6t9l6s33WobyS55qTR5jceMrADQ,3062
 sonusai/aawscd_probwrite.py,sha256=GukR5owp_0A3DrqSl9fHWULYgclNft4D5OkHIwfxxkc,3698
 sonusai/audiofe.py,sha256=3LssRiL73DH8teihD9f3nCvfZ0a65WQtXCqWGnKHuJM,11157
 sonusai/calc_metric_spenh.py,sha256=ee2xrx6L1lFyWSoQSiq56He3RQ1cF7T_ak-6TjejXsc,47738
@@ -44,9 +44,9 @@ sonusai/mixture/db_datatypes.py,sha256=GDYbcSrlgUJsesiUUNnR4s5aBkMgviiNSQDaBcgYX
 sonusai/mixture/eq_rule_is_valid.py,sha256=MpQwRA5M76wSiQWEI1lW2cLFdPaMttBLcQp3tWD8efM,1243
 sonusai/mixture/feature.py,sha256=bHAPRaYGyS-ZTOb-RLCwDau7n1NDKsVEW30Gd9SRZYo,3676
 sonusai/mixture/generation.py,sha256=W3n6ipI-dxg4Wj6YBJn8RTpFqkAyIXzxwObeFbSLq08,42801
-sonusai/mixture/helpers.py,sha256=ijADvIy_xVf0btAsvrQtHv6_6UqDIuesYESp_p3B1KA,24647
+sonusai/mixture/helpers.py,sha256=9x7gezEqPm5xKGAbwCqDMjedVEmoDWyFR_5-T_5nlno,24740
 sonusai/mixture/log_duration_and_sizes.py,sha256=baTUpqyM15wA125jo9E3posmVJUe3WlpksyO6v9Jul0,1347
-sonusai/mixture/mixdb.py,sha256=63LnDBfVKKXSgi825tXCELoQ3xH_jAguW91Xlb86_qE,62807
+sonusai/mixture/mixdb.py,sha256=mr9Ck4p_mCfvz1PXoUgjWcw9F-Rlw3uGiDizUvPqo2A,64359
 sonusai/mixture/soundfile_audio.py,sha256=BwO4lftNvrhoPTJERONcrpxSpM2fjO6kL_e5Ylz742A,4220
 sonusai/mixture/sox_audio.py,sha256=DbHuyLtEuQYtKsIRxx6g1webW_LsdgLz52P5VO37MqI,17119
 sonusai/mixture/sox_augmentation.py,sha256=kBWPrsFk0EBi71nLcKt5v0GA34bY7g9D9x0cEamNWbU,4564
@@ -68,7 +68,6 @@ sonusai/mixture/truth_functions/target.py,sha256=XypzXVMi24Ys13TiEM9JFY_cvHK61Lo
 sonusai/mkwav.py,sha256=zfSyIiQTIK3KV9Ij33jkLhhZIMVYqaROcRQ4S7c4sIo,5364
 sonusai/onnx_predict.py,sha256=jSxhD2oFyGSTHOGCXbW4fRT-k4SqKOboK2JaDO-yWcs,8737
 sonusai/plot.py,sha256=ERkmxMM3qjcCDm4LGDQY4fRAncCYAzP7uW8iZ7_brcg,17105
-sonusai/post_spenh_targetf.py,sha256=MBikRQfVfSZtRz9I5R3muxUtzR83S-i5INu3fAXliT4,4959
 sonusai/queries/__init__.py,sha256=oKY5JeqZ4Cz7DwCwPc1_ydB8bUs6KaMcWFp_w02TjOs,255
 sonusai/queries/queries.py,sha256=oV-m9uiLZOwYTK-Wo7Gf8dpGisaoGf6uDsAJAarVqZI,7553
 sonusai/speech/__init__.py,sha256=SuPcU_K9wQISsZRIzsRNLtEC6cb616l-Jlx3PU-HWMs,113
@@ -120,7 +119,7 @@ sonusai/utils/stratified_shuffle_split.py,sha256=rJNXvBp-GxoKzH3OpL7k0ANSu5xMP2z
 sonusai/utils/write_audio.py,sha256=ZsPGExwM86QHLLN2LOWekK2uAqf5pV_1oRW811p0QAI,840
 sonusai/utils/yes_or_no.py,sha256=eMLXBVH0cEahiXY4W2KNORmwNQ-ba10eRtldh0y4NYg,263
 sonusai/vars.py,sha256=m2AefF0m5bXWGXpJj8Pi42zWL2ydeEj7bkak3GrtMyM,940
-sonusai-0.18.7.dist-info/METADATA,sha256=sVH6EUuzc-eaPUUAPtOqWJrvCYtuGyaSytSdf2q44p0,2591
-sonusai-0.18.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-sonusai-0.18.7.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
-sonusai-0.18.7.dist-info/RECORD,,
+sonusai-0.18.8.dist-info/METADATA,sha256=KqBUJv7yMq-3lDPNfezRqi_z28ZB0w0mDEXJfBtlVmA,2591
+sonusai-0.18.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+sonusai-0.18.8.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
+sonusai-0.18.8.dist-info/RECORD,,

sonusai/post_spenh_targetf.py

@@ -1,160 +0,0 @@
-"""sonusai post_spenh_targetf
-
-usage: post_spenh_targetf [-hv] (-m MODEL) (-w KMODEL) INPUT ...
-
-options:
-    -h, --help
-    -v, --verbose                   Be verbose.
-    -m MODEL, --model MODEL         Python model file.
-    -w KMODEL, --weights KMODEL     Keras model weights file.
-
-Run post-processing on speech enhancement targetf prediction data.
-
-Inputs:
-    MODEL       A SonusAI Python model file with build and/or hypermodel functions.
-    KMODEL      A Keras model weights file (or model file with weights).
-    INPUT       A single H5 file or a glob of H5 files
-
-Outputs the following to post_spenh_targetf-<TIMESTAMP> directory:
-    <name>.wav
-    post_spenh_targetf.log
-
-"""
-import signal
-from dataclasses import dataclass
-
-
-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)
-
-
-@dataclass
-class MPGlobal:
-    N: int = None
-    R: int = None
-    bin_start: int = None
-    bin_end: int = None
-    ttype: str = None
-    output_dir: str = None
-
-
-MP_GLOBAL = MPGlobal()
-
-
-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']
-    model_name = args['--model']
-    weights_name = args['--weights']
-    input_name = args['INPUT']
-
-    import time
-    from os import makedirs
-    from os.path import isfile
-    from os.path import join
-    from os.path import splitext
-
-    from pyaaware import FeatureGenerator
-    from tqdm import tqdm
-
-    from sonusai import create_file_handler
-    from sonusai import initial_log_messages
-    from sonusai import logger
-    from sonusai import update_console_handler
-    from sonusai.utils import create_ts_name
-    from sonusai.utils import import_and_check_keras_model
-    from sonusai.utils import pp_tqdm_imap
-    from sonusai.utils import seconds_to_hms
-
-    start_time = time.monotonic()
-
-    output_dir = create_ts_name('post_spenh_targetf')
-    makedirs(output_dir, exist_ok=True)
-
-    # Setup logging file
-    create_file_handler(join(output_dir, 'post_spenh_targetf.log'))
-    update_console_handler(verbose)
-    initial_log_messages('post_spenh_targetf')
-
-    hypermodel = import_and_check_keras_model(model_name=model_name, weights_name=weights_name)
-
-    fg = FeatureGenerator(feature_mode=hypermodel.feature,
-                          num_classes=hypermodel.num_classes,
-                          truth_mutex=hypermodel.truth_mutex)
-
-    MP_GLOBAL.N = fg.itransform_N
-    MP_GLOBAL.R = fg.itransform_R
-    MP_GLOBAL.bin_start = fg.bin_start
-    MP_GLOBAL.bin_end = fg.bin_end
-    MP_GLOBAL.ttype = fg.itransform_ttype
-    MP_GLOBAL.output_dir = output_dir
-
-    if not all(isfile(file) and splitext(file)[1] == '.h5' for file in input_name):
-        logger.exception(f'Do not know how to process input from {input_name}')
-        raise SystemExit(1)
-
-    logger.info('')
-    logger.info(f'Found {len(input_name):,} files to process')
-
-    progress = tqdm(total=len(input_name))
-    pp_tqdm_imap(_process, input_name, progress=progress)
-    progress.close()
-
-    logger.info(f'Wrote {len(input_name)} mixtures to {output_dir}')
-    logger.info('')
-
-    end_time = time.monotonic()
-    logger.info(f'Completed in {seconds_to_hms(seconds=end_time - start_time)}')
-    logger.info('')
-
-
-def _process(file: str) -> None:
-    """Run extraction on predict data to generate estimation audio
-    """
-    from os.path import basename
-    from os.path import join
-    from os.path import splitext
-
-    import h5py
-    import numpy as np
-    from sonusai import InverseTransform
-
-    from sonusai import SonusAIError
-    from sonusai.mixture import get_audio_from_transform
-    from sonusai.utils import float_to_int16
-    from sonusai.utils import unstack_complex
-    from sonusai.utils import write_audio
-
-    try:
-        with h5py.File(file, 'r') as f:
-            predict = unstack_complex(np.array(f['predict']))
-    except Exception as e:
-        raise SonusAIError(f'Error reading {file}: {e}')
-
-    output_name = join(MP_GLOBAL.output_dir, splitext(basename(file))[0] + '.wav')
-    audio, _ = get_audio_from_transform(data=predict,
-                                        transform=InverseTransform(N=MP_GLOBAL.N,
-                                                                   R=MP_GLOBAL.R,
-                                                                   bin_start=MP_GLOBAL.bin_start,
-                                                                   bin_end=MP_GLOBAL.bin_end,
-                                                                   ttype=MP_GLOBAL.ttype,
-                                                                   gain=np.float32(1)))
-    write_audio(name=output_name, audio=float_to_int16(audio))
-
-
-if __name__ == '__main__':
-    main()