sonusai 0.15.8__py3-none-any.whl → 0.15.9__py3-none-any.whl

sonusai/utils/audio_devices.py

@@ -0,0 +1,41 @@
+import pyaudio
+
+
+def get_input_device_index_by_name(p: pyaudio.PyAudio, name: str = None) -> int:
+    info = p.get_host_api_info_by_index(0)
+    device_count = info.get('deviceCount')
+    for i in range(0, device_count):
+        device_info = p.get_device_info_by_host_api_device_index(0, i)
+        if name is None:
+            device_name = None
+        else:
+            device_name = device_info.get('name')
+        if name == device_name and device_info.get('maxInputChannels') > 0:
+            return i
+
+    raise ValueError(f'Could not find {name}')
+
+
+def get_input_devices(p: pyaudio.PyAudio) -> list[str]:
+    names = []
+    info = p.get_host_api_info_by_index(0)
+    device_count = info.get('deviceCount')
+    for i in range(0, device_count):
+        device_info = p.get_device_info_by_host_api_device_index(0, i)
+        device_name = device_info.get('name')
+        if device_info.get('maxInputChannels') > 0:
+            names.append(device_name)
+
+    return names
+
+
+def get_default_input_device(p: pyaudio.PyAudio) -> str:
+    info = p.get_host_api_info_by_index(0)
+    device_count = info.get('deviceCount')
+    for i in range(0, device_count):
+        device_info = p.get_device_info_by_host_api_device_index(0, i)
+        device_name = device_info.get('name')
+        if device_info.get('maxInputChannels') > 0:
+            return device_name
+
+    raise ValueError('No input audio devices found')

sonusai/utils/calculate_input_shape.py

@@ -13,13 +13,12 @@ def calculate_input_shape(feature: str,
     """
     from pyaaware import FeatureGenerator
 
-    # num_classes is irrelevant, set to 2
-    fg = FeatureGenerator(feature_mode=feature, num_classes=2)
+    fg = FeatureGenerator(feature_mode=feature)
 
     if flatten:
-        in_shape = [fg.stride * fg.num_bands]
+        in_shape = [fg.stride * fg.feature_parameters]
     else:
-        in_shape = [fg.stride, fg.num_bands]
+        in_shape = [fg.stride, fg.feature_parameters]
 
     if timesteps > 0:
         in_shape.insert(0, timesteps)

sonusai/utils/create_timestamp.py

@@ -0,0 +1,5 @@
+def create_timestamp() -> str:
+    """Create a timestamp."""
+    from datetime import datetime
+
+    return datetime.now().strftime('%Y%m%d-%H%M%S')

sonusai/utils/reshape.py

@@ -17,14 +17,14 @@ def reshape_inputs(feature: Feature,
                    timesteps: int = 0,
                    flatten: bool = False,
                    add1ch: bool = False) -> tuple[Feature, Optional[Truth]]:
-    """Check SonusAI feature and truth data and reshape feature of size [frames, strides, bands] into
+    """Check SonusAI feature and truth data and reshape feature of size [frames, strides, feature_parameters] into
     one of several options:
 
     If timesteps > 0: (i.e., for recurrent NNs):
-      no-flatten, no-channel:   [sequences, timesteps, strides, bands]      (4-dim)
-      flatten, no-channel:      [sequences, timesteps, strides*bands]       (3-dim)
-      no-flatten, add-1channel: [sequences, timesteps, strides, bands, 1]   (5-dim)
-      flatten, add-1channel:    [sequences, timesteps, strides*bands, 1]    (4-dim)
+      no-flatten, no-channel:   [sequences, timesteps, strides, feature_parameters]      (4-dim)
+      flatten, no-channel:      [sequences, timesteps, strides*feature_parameters]       (3-dim)
+      no-flatten, add-1channel: [sequences, timesteps, strides, feature_parameters, 1]   (5-dim)
+      flatten, add-1channel:    [sequences, timesteps, strides*feature_parameters, 1]    (4-dim)
 
     If batch_size is None, then do not reshape; just calculate new input shape and return.
 
@@ -40,7 +40,7 @@ def reshape_inputs(feature: Feature,
     """
     from sonusai import SonusAIError
 
-    frames, strides, bands = feature.shape
+    frames, strides, feature_parameters = feature.shape
     if truth is not None:
         truth_frames, num_classes = truth.shape
         # Double-check correctness of inputs
@@ -50,7 +50,7 @@ def reshape_inputs(feature: Feature,
         num_classes = None
 
     if flatten:
-        feature = np.reshape(feature, (frames, strides * bands))
+        feature = np.reshape(feature, (frames, strides * feature_parameters))
 
     # Reshape for Keras/TF recurrent models that require timesteps/sequence length dimension
     if timesteps > 0:
@@ -73,14 +73,14 @@ def reshape_inputs(feature: Feature,
 
         # Reshape
         if feature.ndim == 2:  # flattened input
-            # was [frames, bands*timesteps]
-            feature = np.reshape(feature, (sequences, timesteps, strides * bands))
+            # was [frames, feature_parameters*timesteps]
+            feature = np.reshape(feature, (sequences, timesteps, strides * feature_parameters))
             if truth is not None:
                 # was [frames, num_classes]
                 truth = np.reshape(truth, (sequences, timesteps, num_classes))
         elif feature.ndim == 3:  # un-flattened input
-            # was [frames, bands, timesteps]
-            feature = np.reshape(feature, (sequences, timesteps, strides, bands))
+            # was [frames, feature_parameters, timesteps]
+            feature = np.reshape(feature, (sequences, timesteps, strides, feature_parameters))
             if truth is not None:
                 # was [frames, num_classes]
                 truth = np.reshape(truth, (sequences, timesteps, num_classes))

sonusai/utils/wave.py

@@ -5,15 +5,22 @@ from sonusai.mixture.datatypes import AudioT
 def write_wav(name: str, audio: AudioT, sample_rate: int = SAMPLE_RATE) -> None:
     """ Write a simple, uncompressed WAV file.
 
-    To write multiple channels, use a 2D array of shape [samples, channels].
+    To write multiple channels, use a 2D array of shape [channels, samples].
     The bits per sample and PCM/float are determined by the data type.
 
     """
-    import numpy as np
     import torch
     import torchaudio
 
-    if audio.ndim == 1:
-        audio = np.reshape(audio, (1, audio.shape[0]))
+    data = torch.tensor(audio)
 
-    torchaudio.save(name, torch.tensor(audio), sample_rate)
+    if data.dim() == 1:
+        data = torch.reshape(data, (1, data.shape[0]))
+    if data.dim() != 2:
+        raise ValueError(f'audio must be a 1D or 2D array')
+
+    # Assuming data has more samples than channels, check if array needs to be transposed
+    if data.shape[1] < data.shape[0]:
+        data = torch.transpose(data, 0, 1)
+
+    torchaudio.save(uri=name, src=data, sample_rate=sample_rate)

{sonusai-0.15.8.dist-info → sonusai-0.15.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sonusai
-Version: 0.15.8
+Version: 0.15.9
 Summary: Framework for building deep neural network models for sound, speech, and voice AI
 Home-page: https://aaware.com
 License: GPL-3.0-only
@@ -16,28 +16,35 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: PyYAML (>=6.0.1,<7.0.0)
 Requires-Dist: aixplain (>=0.2.6,<0.3.0)
+Requires-Dist: bitarray (>=2.9.2,<3.0.0)
 Requires-Dist: ctranslate2 (==4.1.0)
 Requires-Dist: dataclasses-json (>=0.6.1,<0.7.0)
 Requires-Dist: deepgram-sdk (>=3.0.0,<4.0.0)
 Requires-Dist: docopt (>=0.6.2,<0.7.0)
+Requires-Dist: einops (>=0.7.0,<0.8.0)
 Requires-Dist: faster-whisper (>=1.0.1,<2.0.0)
+Requires-Dist: geomloss (>=0.2.6,<0.3.0)
 Requires-Dist: h5py (>=3.11.0,<4.0.0)
+Requires-Dist: hydra-core (>=1.3.2,<2.0.0)
 Requires-Dist: jiwer (>=3.0.3,<4.0.0)
 Requires-Dist: keras (>=3.1.1,<4.0.0)
 Requires-Dist: keras-tuner (>=1.4.7,<2.0.0)
 Requires-Dist: librosa (>=0.10.1,<0.11.0)
 Requires-Dist: lightning (>=2.2,<2.3)
 Requires-Dist: matplotlib (>=3.8.0,<4.0.0)
+Requires-Dist: omegaconf (>=2.3.0,<3.0.0)
 Requires-Dist: onnx (>=1.14.1,<2.0.0)
 Requires-Dist: onnxruntime (>=1.16.1,<2.0.0)
 Requires-Dist: paho-mqtt (>=2.0.0,<3.0.0)
 Requires-Dist: pandas (>=2.1.1,<3.0.0)
 Requires-Dist: pesq (>=0.0.4,<0.0.5)
 Requires-Dist: pyaaware (>=1.5.3,<2.0.0)
+Requires-Dist: pyaudio (>=0.2.14,<0.3.0)
 Requires-Dist: pydub (>=0.25.1,<0.26.0)
 Requires-Dist: pystoi (>=0.4.0,<0.5.0)
 Requires-Dist: python-magic (>=0.4.27,<0.5.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)
+Requires-Dist: sacrebleu (>=2.4.2,<3.0.0)
 Requires-Dist: samplerate (>=0.2.1,<0.3.0)
 Requires-Dist: soundfile (>=0.12.1,<0.13.0)
 Requires-Dist: sox (>=1.4.1,<2.0.0)

{sonusai-0.15.8.dist-info → sonusai-0.15.9.dist-info}/RECORD

@@ -1,27 +1,27 @@
 sonusai/__init__.py,sha256=KmIJ9wni9d9v5pyu0pUxbacZIHGkAywB9CJwl7JME28,1526
 sonusai/aawscd_probwrite.py,sha256=GukR5owp_0A3DrqSl9fHWULYgclNft4D5OkHIwfxxkc,3698
-sonusai/calc_metric_spenh.py,sha256=Cf02uYB6fzDX5anQuBTR_aD7mYt1WgxkEIzZA08uFvs,60825
+sonusai/audiofe.py,sha256=XE_cgOhhTryjPUePxW_8NY1TwrnRZ6BHCsH-gp8PmYw,11471
+sonusai/calc_metric_spenh.py,sha256=D8iQVSIhFhrsUwKuIP-S38NBnyfAOZlsOIIgOZwGOOI,60852
 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
 sonusai/data/whitenoise.wav,sha256=I2umov0m34y56F9IsIBi1XtE76ZeZaSKDf70cJRe3pI,1920044
 sonusai/data_generator/__init__.py,sha256=ouCpY5EDV35fKFeKGQfIcU8uE-c3QcuNerTxUA1X5L8,232
-sonusai/data_generator/dataset_from_mixdb.py,sha256=Fe_7xxCYURvbZjuIZrvzozdkrx7ZO7gxGg8c5ob2dys,5478
-sonusai/data_generator/keras_from_mixdb.py,sha256=V5CUsGz-akIYdgQy9ABxwNKMYKv01klA4xtMDveF6uI,6167
+sonusai/data_generator/dataset_from_mixdb.py,sha256=D14L8BL7a0WgkF8a8eogQ9Hk9ow4_RK3QBGsZ-HDAog,5493
+sonusai/data_generator/keras_from_mixdb.py,sha256=14r89aX6Dr9ZKsmMRC7HDXbJrPrCZC1liwwLmZUKj0w,6182
 sonusai/data_generator/torch_from_mixdb.py,sha256=lvEe9DDu_rIaoyhv9PW4UAnAWp5N74L8kRfxUhsh7oo,4279
 sonusai/doc/__init__.py,sha256=rP5Hgn0Iys_xkuv4caxngdqehuU4zLZsiKuv8Nde67M,19
 sonusai/doc/doc.py,sha256=3z210v6ZckuOlsGZ3ySQBdlCNmBp2M1ahqhqG_eUN58,22664
 sonusai/doc.py,sha256=l8CaFgLI8mqx4tn0aXfxKqa2dy9GgC0zjYxZAkpmi1E,878
-sonusai/evaluate.py,sha256=1BX9dCXMHg6LefbvkevxYpjM3JR-i0jR3Eob2oNh_hQ,9983
-sonusai/genft.py,sha256=CeQN8Sxb_NKeXWJxN9HtzUu687eXl97nHBxzzCzQdLg,5557
+sonusai/genft.py,sha256=6hOds8d-pYRawesLYh7XLrQh4PweWUj8U5Cbzx45bNQ,5572
 sonusai/genmix.py,sha256=0AiUfF7n0CGOp5v-woNfeP3-QaVQUb0CJZc0oXkvqpk,7016
-sonusai/genmixdb.py,sha256=onWkaCPzmUMvDtiGf3A7UdGkOA8xe9zuJTsuLfMdb_s,19597
+sonusai/genmixdb.py,sha256=rAxCKNPkOXaAugEfp9pTcHCQONapdTnxMlBsIPIoizE,19639
 sonusai/gentcst.py,sha256=8jeXirgJe0OGgknC8A-rIudjHeH8UTYPpuh71Ha-I3w,20165
 sonusai/keras_onnx.py,sha256=WHcPIcff4VPdiXqGX-TU-_x-UuKUD3nNpQtCX-2NEVQ,2658
-sonusai/keras_predict.py,sha256=ffrtI12T8_Rm26KsBee0dfEcnEuhOxvtV80YmbJR0m0,9085
+sonusai/keras_predict.py,sha256=_83EtPtnfrqwUzC2H2tk4LI90RiQdyEEBxFGTgFPl3M,9090
 sonusai/keras_train.py,sha256=8_M5vY-CkonPzbOtOF3Vk-wox-42o8fkaOKLjk7Oc2k,13226
-sonusai/lsdb.py,sha256=qTEHQ5X9Ruc0ph4OUhYKFZ6xGC3gyhcNq7t25oyLhc8,5929
-sonusai/main.py,sha256=_s3WCFP_zQ7pD0OgdSVRPQRE38V5Qp6bG56JX-Y79ME,3120
+sonusai/lsdb.py,sha256=TTMQ-0H8fFzUSczt6yjy-9xUjZSdIGQzTVH5Xr6XPSA,5941
+sonusai/main.py,sha256=KjN0dCI6rWare4wo_ACzTlURW7pvTw03n51pH7EyLAU,3108
 sonusai/metrics/__init__.py,sha256=56itZW3S1I7ZYvbxPmFIVPAh1AIJZdljByz1uCrHqFE,635
 sonusai/metrics/calc_class_weights.py,sha256=dyY7daEIf5Ms5tfTf6wF0fkx_GnMADHOZR_rtsfGoVM,3933
 sonusai/metrics/calc_optimal_thresholds.py,sha256=9fRfwl-aKAbzHJyqGHv4o8BpZXG9HHB7zUJObHXfYM4,3522
@@ -35,24 +35,24 @@ sonusai/metrics/class_summary.py,sha256=4Mb25nuk6eqotnQSFMuOQL3zofGcpNXDfDlPa513
 sonusai/metrics/confusion_matrix_summary.py,sha256=3qg6TMKjJeHtNjj2YnNjPFSlMrQXt0Zcu1dLkGB_aPU,4001
 sonusai/metrics/one_hot.py,sha256=QSeH_GdqBpOAKLrNnQ8gjcPC-vSdUqC0yPEQueTA6VI,13548
 sonusai/metrics/snr_summary.py,sha256=P4U5_Xr7v9F8kF-rZBnpsVNt3p42rIVS6zmch8yfVfg,5575
-sonusai/mixture/__init__.py,sha256=WzPHGSWz6v64HCSFTGboG5o-xBy_0II4i4tkf1UL1Vw,5251
-sonusai/mixture/audio.py,sha256=6W2ihjJGBy7Xggx1imF7bzkymDuipPOgC63j2J7Wf-E,3456
+sonusai/mixture/__init__.py,sha256=fCVSlizYxUUQQD9nSZ8bEbfc_TB2yiOC14HPOB4KFz4,5287
+sonusai/mixture/audio.py,sha256=S-ZROf5rVvwv1TCEuwJHz1FfX4oVubb4QhbybUMMqtM,2150
 sonusai/mixture/augmentation.py,sha256=Blb90tdTwBOj5w9tRcYyS5H67YJuFiXsGqwZWd7ON4g,10468
 sonusai/mixture/class_count.py,sha256=_wFnVl2yEOnbor7pLg7cYOUeX6nioov-03Cv3SEbh2k,996
 sonusai/mixture/config.py,sha256=CXIkVRJmaW2QW_sGl0aIqPf7I_TesyGhUYzxouw5UX4,22266
 sonusai/mixture/constants.py,sha256=xjCskcQi6khqYZDf7j6z1OkeN1C6wE06kBBapcJiNI4,1428
-sonusai/mixture/datatypes.py,sha256=xN-GdPCEHGE2Ak_TdFbjuSyMs4x7TLRp59trbMTiYLg,8164
+sonusai/mixture/datatypes.py,sha256=zaxfOHw8ddt-i8JPYOPnlqWz_EHBEDoO4q2VAqJViHM,8173
 sonusai/mixture/eq_rule_is_valid.py,sha256=MpQwRA5M76wSiQWEI1lW2cLFdPaMttBLcQp3tWD8efM,1243
-sonusai/mixture/feature.py,sha256=io6OiJAJ3GYvPChiUmPQuP3h0OB2onjYF8o9-AWkmqM,1996
+sonusai/mixture/feature.py,sha256=Rwuf82IoXzhHPGbKYVGcatImF_ssBf_FfvbqghVPXtg,4116
 sonusai/mixture/generation.py,sha256=miUrc3QOSUNIG6mDkiMCZ6M2ulivUZxlYUAJUOVomWc,39039
-sonusai/mixture/helpers.py,sha256=XqpcB15MezEMVJwf3jxzATDJSpj_27b8Cru1TDIFD7w,21326
-sonusai/mixture/log_duration_and_sizes.py,sha256=r-wVjrLW1XBciOL4pkZSYMR7ZNADbojE95TPSQkp3kc,1329
+sonusai/mixture/helpers.py,sha256=GSGSD2KnvOeEIB6IwNTxyaQNjghTSBMB729kUEd_RiM,22403
+sonusai/mixture/log_duration_and_sizes.py,sha256=baTUpqyM15wA125jo9E3posmVJUe3WlpksyO6v9Jul0,1347
 sonusai/mixture/mapped_snr_f.py,sha256=mlbYM1t14OXe_Zg4CjpWTuA_Zun4W0O3bSUXeodRBQs,1845
-sonusai/mixture/mixdb.py,sha256=FQ5hirb2zR8Aj1UNtz89qJQ8wlE0ELC80IxQDmyhsKk,45188
+sonusai/mixture/mixdb.py,sha256=9Pe0mEG8pnEf9NZynTIldc05GfdOrgmcVoIt63RG5DA,45279
 sonusai/mixture/soundfile_audio.py,sha256=Ow_IWIMz4pMsLxMP_JsQ8AuHLCWlYQinLa58CFW97f8,2804
 sonusai/mixture/sox_audio.py,sha256=HT3kYA9TP5QPCuoOJdUMnGVN-qY6q96DGL8zxuog76o,12277
 sonusai/mixture/sox_augmentation.py,sha256=F9tBdNvX2guCn7gRppAFrxRnBtjw9q6qAq2_v_A4hh0,4490
-sonusai/mixture/spectral_mask.py,sha256=LKFnrqZryPHT6FBNiT7yFxOeXc6-AUg6X54N26d8ctg,2107
+sonusai/mixture/spectral_mask.py,sha256=8AkCwhy-PSdP1Uri9miKZP-bXFYnFcH_c9xZCGrHavU,2071
 sonusai/mixture/target_class_balancing.py,sha256=NTNiKZH0_PWLooeow0l41CjJKK8ZTMVbUqz9ZkaNtWk,4900
 sonusai/mixture/targets.py,sha256=wyy5vhLhuN-hqBMBGoziVvEJg3FKFvJFgmEE7_LaV2M,7908
 sonusai/mixture/tokenized_shell_vars.py,sha256=gCxw8SQUcal6mqWKF7hOBTgSQmbJUk1nT0Gn3H8GA0U,4705
@@ -61,24 +61,24 @@ sonusai/mixture/torchaudio_augmentation.py,sha256=1vEDHI0caL1vrgoY2lAWe4CiHE2jKR
 sonusai/mixture/truth.py,sha256=Y41pZ52Xkols9LUler0NlgnilUOscBIucmw4GcxXNzU,1612
 sonusai/mixture/truth_functions/__init__.py,sha256=82lKYHhLy8KW3gHngrocoqwupGVLVsWdIXdYs3vhjOc,359
 sonusai/mixture/truth_functions/crm.py,sha256=_Vy8UMrOUQXsrM3nutvUMWCpvI8GePr01QFlyqLFd4k,2626
-sonusai/mixture/truth_functions/data.py,sha256=NJNZz5fB3jnntUDlnsKJVQIeuHNUvD4x5iNaQVQlo3Y,2857
+sonusai/mixture/truth_functions/data.py,sha256=okFJeOf43NxfdLqWFCBA2pOGqujRlNDYdAcwwR_m8z8,2875
 sonusai/mixture/truth_functions/energy.py,sha256=ydMtMLjMloG76DB30ZHQ5tkBVh4dkMJ82XEhKBokmIk,4281
 sonusai/mixture/truth_functions/file.py,sha256=jOJuC_3y9BH6GGOp9eKcbVrHLVRzUA80BJq59LhcBUM,1539
 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=3rPXYwU4SBiPP3uIDpOL-B2Xw1Zh3JboD_MYNEyUpuk,5746
+sonusai/mixture/truth_functions/target.py,sha256=KAsjugDRooOA5BRcHVAbZRgV7l8S5CFg7CZ0XtKZaQ0,5764
 sonusai/mkmanifest.py,sha256=dIPVFKKhnhHdq63OGr6p__pK7fyx3OdKVtbmGUJxsR8,7078
 sonusai/mkwav.py,sha256=LZNyhq4gJEs_NtGvRsYHA2qfgkkODpt6HoH1b-Tjjuw,5266
-sonusai/onnx_predict.py,sha256=RhQbbNG3w6rCXuSFUWCaQmUH5JzSP2hmu6TG5_81IVA,9055
-sonusai/plot.py,sha256=GPrbk7SxwojtjR9KLE9jQ6ywYWXca296W0iaRPdnBHI,16982
-sonusai/post_spenh_targetf.py,sha256=Y9J5JbFnCFKCednNiYvVXaC5Z3lF5KxifFJ2RCy4jmg,4975
+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/queries/__init__.py,sha256=oKY5JeqZ4Cz7DwCwPc1_ydB8bUs6KaMcWFp_w02TjOs,255
 sonusai/queries/queries.py,sha256=FNMUKnoY_Ya9S5sNhsB8ppwy0B7V55ilbbjhQRv_UN8,7552
 sonusai/torchl_onnx.py,sha256=5JYow3XpBaUdtuyAW0mOZyCKL_4FrHvEekYBRdDT6KA,8967
-sonusai/torchl_predict.py,sha256=-wlUdRGPjOvGjCQZY277D8tGeZp1KyDU7TpYH7ovC0c,26657
+sonusai/torchl_predict.py,sha256=P1ySDH_ITOPefZ2xZqyxyIrsNDqblKTBLZqFApgo5EU,26238
 sonusai/torchl_train.py,sha256=NPCRB0gwTvabivmOz78gjUreDeO1z16PYuw7L1-pIRQ,9680
 sonusai/tplot.py,sha256=yFyyyg9ymp2Eh-64Muu0EFFEY61MoJSV0a_fy9OWaCk,14485
-sonusai/utils/__init__.py,sha256=ynmdoPJVdbk5eYq9TA9t4QpGEkdiGATDx1pFu43u7YQ,2180
+sonusai/utils/__init__.py,sha256=tVSmxinSo0Enexpol6wCzz6tU7WrueC-YslFgQr-o7M,2382
 sonusai/utils/asl_p56.py,sha256=GCKlz-NLInQ0z41XBi0mOvGdSfRZf3WI53necVNDo80,3837
 sonusai/utils/asr.py,sha256=QN1wdO9-EqD72-ixr4lnzsPfT8i0syhTGj1evKNJWe4,2021
 sonusai/utils/asr_functions/__init__.py,sha256=4boXXOXlQHTt8K2DWOwFXSlc8D2NLFd8QTc68yL2ejU,214
@@ -93,9 +93,11 @@ sonusai/utils/asr_manifest_functions/__init__.py,sha256=Lz12aCGvfngZkLoUxHSqFjHc
 sonusai/utils/asr_manifest_functions/data.py,sha256=mJsaHccBReguOJu9qsshRhL-3GbeyqM0-PXMseFnZbE,151
 sonusai/utils/asr_manifest_functions/librispeech.py,sha256=HIaytcYmjRUkuR6fCQlv3Jh3IDWSox_A6WFcFFAHN9M,1635
 sonusai/utils/asr_manifest_functions/vctk_noisy_speech.py,sha256=-69lM0dz18KbU5_-dmSeqDoNNwgJj4UlxgGkNBEi3wM,2169
+sonusai/utils/audio_devices.py,sha256=LgaXTln1oRArBzaet3rZiIO2plgtaThuGBc3sJ_sLlo,1414
 sonusai/utils/braced_glob.py,sha256=h4hab7YDbM4CjLg9iSzyHZrkd22IPUOY5zZqHdifkh8,1510
-sonusai/utils/calculate_input_shape.py,sha256=fyf8Ggxn9xljJ87BNwIzOh2KD4pKUGFZ2RNzE3195NI,1023
+sonusai/utils/calculate_input_shape.py,sha256=63ILxibYKuTQozY83QN8Y2OOhBEbW_1X47Q0askcHDM,984
 sonusai/utils/convert_string_to_number.py,sha256=i17yIxurp8Iz6NPE-imTRlARrXWqadwm8qbOTuzHZvE,236
+sonusai/utils/create_timestamp.py,sha256=TxoQXWZ3SFdBEHLOv-ujeIsTEJuiFnKOGRy-FQq45YU,148
 sonusai/utils/create_ts_name.py,sha256=8RLKmgXwuGcbDMGgtTuc0MvGFfA7IOVqfjkE2T18GOo,405
 sonusai/utils/dataclass_from_dict.py,sha256=vAGnuMjhy0W9bxZ5usrH7mbQsFog3n0__IC4xyJyVUc,390
 sonusai/utils/db.py,sha256=lI77MJJLs4CTYxhjFUvBom2Kk2imAP34okOeO4irbDc,371
@@ -114,15 +116,15 @@ sonusai/utils/print_mixture_details.py,sha256=BzYM4-wHHNa6zxPzBMUJxwKt0gKHmvbwdd
 sonusai/utils/ranges.py,sha256=NPBZOVzMb95GTOIxltVO-wSzgcXqZ14wbdV46JDLKrw,1222
 sonusai/utils/read_mixture_data.py,sha256=Sb30RgSpw6DnH_iD81O7G_KOsdfjQWWLk3euEkxfMa8,453
 sonusai/utils/read_predict_data.py,sha256=5rR_ijrrcS2cKO1Sea2M2QEicokTtW5XtAo6jT5YSX8,1064
-sonusai/utils/reshape.py,sha256=sz-V4Za3DIvqFztTh0yYhQRcofnjc9XBW0CBhxg18lo,5854
+sonusai/utils/reshape.py,sha256=E8Eu6grynaeWwVO6peIR0BF22SrVaJSa1Rkl109lq6Y,5997
 sonusai/utils/seconds_to_hms.py,sha256=oxLuZhTJJr9swj-fOSOrZJ5vBNM7_BrOMQhX1pYpiv0,260
 sonusai/utils/stacked_complex.py,sha256=feLhz3GC1ILxBGMHOj3sJK--sidsXKbfwkalwAVwizc,2950
 sonusai/utils/stratified_shuffle_split.py,sha256=rJNXvBp-GxoKzH3OpL7k0ANSu5xMP2zJ7K1fm_33UzE,7022
 sonusai/utils/trim_docstring.py,sha256=dSrtiRsEN4wkkvKBp6WDr13RUypfqZzgH_jOBLs1ouY,881
-sonusai/utils/wave.py,sha256=TKE-CNPGFXNXUW626CBPzCTNgWJut8I0ZEUsgG9q4Po,586
+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.15.8.dist-info/METADATA,sha256=3eCpCJmXOfr7GV3a7HDWo0iilEVHB5ANdQqS59O0Yi0,2920
-sonusai-0.15.8.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-sonusai-0.15.8.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
-sonusai-0.15.8.dist-info/RECORD,,
+sonusai-0.15.9.dist-info/METADATA,sha256=DudNQlTEQpWpzqyzyowz_V-J9epd7mrKgAYM6rFxaPo,3209
+sonusai-0.15.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+sonusai-0.15.9.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
+sonusai-0.15.9.dist-info/RECORD,,

{sonusai-0.15.8.dist-info → sonusai-0.15.9.dist-info}/WHEEL

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

sonusai/evaluate.py

@@ -1,245 +0,0 @@
-"""sonusai evaluate
-
-usage: evaluate [-hv] [-i MIXID] (-f FEATURE) (-p PREDICT) [-t PTHR] LOC
-
-options:
-    -h, --help
-    -v, --verbose                   Be verbose.
-    -i MIXID, --mixid MIXID         Mixture ID(s) to generate. [default: *].
-    -p PREDICT, --predict PREDICT   A directory containing prediction data.
-    -t PTHR, --thr PTHR             Optional prediction decision threshold(s). [default: 0].
-
-Evaluate calculates performance metrics of neural-network models from model prediction data and genft data.
-
-Inputs:
-    LOC         A SonusAI mixture database directory.
-    MIXID       A glob of mixture ID(s) to generate.
-    PREDICT     A directory containing SonusAI predict HDF5 files. Contains:
-                    dataset:    predict (either [frames, num_classes] or [frames, timesteps, num_classes])
-    PTHR        Scalar or array of thresholds. Default 0 will select values:
-                    argmax()    if mixdb indicates single-label mode (truth_mutex = true)
-                    0.5         if mixdb indicates multi-label mode  (truth_mutex = false)
-                If PTHR = -1, optimal thresholds are calculated using precision_recall_curve() which
-                optimizes F1 score.
-"""
-import numpy as np
-
-from sonusai import logger
-from sonusai.mixture import Feature
-from sonusai.mixture import MixtureDatabase
-from sonusai.mixture import Predict
-from sonusai.mixture import Segsnr
-from sonusai.mixture import Truth
-
-
-def evaluate(mixdb: MixtureDatabase,
-             truth: Truth,
-             predict: Predict = None,
-             segsnr: Segsnr = None,
-             output_dir: str = None,
-             predict_thr: float | np.ndarray = 0,
-             feature: Feature = None,
-             verbose: bool = False) -> None:
-    from os.path import join
-
-    from sonusai import initial_log_messages
-    from sonusai import update_console_handler
-    from sonusai.metrics import calc_optimal_thresholds
-    from sonusai.metrics import class_summary
-    from sonusai.metrics import snr_summary
-    from sonusai.mixture import SAMPLE_RATE
-    from sonusai.queries import get_mixids_from_snr
-    from sonusai.utils import get_num_classes_from_predict
-    from sonusai.utils import human_readable_size
-    from sonusai.utils import reshape_outputs
-    from sonusai.utils import seconds_to_hms
-
-    update_console_handler(verbose)
-    initial_log_messages('evaluate')
-
-    if truth.shape[-1] != predict.shape[-1]:
-        logger.exception(f'Number of classes in truth and predict are not equal. Exiting.')
-        raise SystemExit(1)
-
-    # truth, predict can be either [frames, num_classes] or [frames, timesteps, num_classes]
-    # in binary case dim may not exist, detect this and set num_classes == 1
-    timesteps = -1
-    predict, truth = reshape_outputs(predict=predict, truth=truth, timesteps=timesteps)
-    num_classes = get_num_classes_from_predict(predict=predict, timesteps=timesteps)
-
-    fdiff = truth.shape[0] - predict.shape[0]
-    if fdiff > 0:
-        # truth = truth[0:-fdiff,:]
-        predict = np.concatenate((predict, np.zeros((fdiff, num_classes), dtype=np.float32)))
-        logger.info(f'Truth has more feature-frames than predict, padding predict with zeros to match.')
-
-    if fdiff < 0:
-        predict = predict[0:fdiff, :]
-        logger.info(f'Predict has more feature-frames than truth, trimming predict to match.')
-
-    # Check segsnr, input is always in transform frames
-    compute_segsnr = False
-    if len(segsnr) > 0:
-        segsnr_feature_frames = segsnr.shape[0] / (mixdb.feature_step_samples / mixdb.ft_config.R)
-        if segsnr_feature_frames == truth.shape[0]:
-            compute_segsnr = True
-        else:
-            logger.warning('segsnr length does not match truth, ignoring.')
-
-    # Check predict_thr array or scalar and return final scalar predict_thr value
-    if not mixdb.truth_mutex:
-        if num_classes > 1:
-            if not isinstance(predict_thr, np.ndarray):
-                if predict_thr == 0:
-                    # multi-label predict_thr scalar 0 force to 0.5 default
-                    predict_thr = np.atleast_1d(0.5)
-                else:
-                    predict_thr = np.atleast_1d(predict_thr)
-            else:
-                if predict_thr.ndim == 1:
-                    if predict_thr[0] == 0:
-                        # multi-label predict_thr array scalar 0 force to 0.5 default
-                        predict_thr = np.atleast_1d(0.5)
-                    else:
-                        # multi-label predict_thr array set to scalar = array[0]
-                        predict_thr = predict_thr[0]
-    else:
-        # single-label mode, force argmax mode
-        predict_thr = np.atleast_1d(0)
-
-    if predict_thr == -1:
-        thrpr, thrroc, _, _ = calc_optimal_thresholds(truth, predict, timesteps)
-        predict_thr = np.atleast_1d(thrpr)
-        predict_thr = np.maximum(predict_thr, 0.001)  # enforce lower limit
-        predict_thr = np.minimum(predict_thr, 0.999)  # enforce upper limit
-        predict_thr = predict_thr.round(2)
-
-    # Summarize the mixture data
-    num_mixtures = mixdb.num_mixtures
-    total_samples = sum([mixture.samples for mixture in mixdb.mixtures])
-    duration = total_samples / SAMPLE_RATE
-
-    logger.info('')
-    logger.info(f'Mixtures: {num_mixtures}')
-    logger.info(f'Duration: {seconds_to_hms(seconds=duration)}')
-    logger.info(f'truth:    {human_readable_size(truth.nbytes, 1)}')
-    logger.info(f'predict:  {human_readable_size(predict.nbytes, 1)}')
-    if compute_segsnr:
-        logger.info(f'segsnr:   {human_readable_size(segsnr.nbytes, 1)}')
-    if feature:
-        logger.info(f'feature:  {human_readable_size(feature.nbytes, 1)}')
-
-    logger.info(f'Classes: {num_classes}')
-    if mixdb.truth_mutex:
-        logger.info(f'Mode:  Single-label / truth_mutex / softmax')
-    else:
-        logger.info(f'Mode:  Multi-label / Binary')
-
-    mxid_snro = get_mixids_from_snr(mixdb=mixdb)
-    snrlist = list(mxid_snro.keys())
-    snrlist.sort(reverse=True)
-    logger.info(f'Ordered SNRs: {snrlist}\n')
-    predict_thr_info = predict_thr.transpose() if isinstance(predict_thr, np.ndarray) else predict_thr
-    logger.info(f'Prediction Threshold(s): {predict_thr_info}\n')
-
-    # Top-level report over all mixtures
-    macrodf, microdf, wghtdf, mxid_snro = snr_summary(mixdb=mixdb,
-                                                      mixid=':',
-                                                      truth_f=truth,
-                                                      predict=predict,
-                                                      segsnr=segsnr if compute_segsnr else None,
-                                                      predict_thr=predict_thr)
-
-    if num_classes > 1:
-        logger.info(f'Metrics micro-avg per SNR over all {num_mixtures} mixtures:')
-    else:
-        logger.info(f'Metrics per SNR over all {num_mixtures} mixtures:')
-    logger.info(microdf.round(3).to_string())
-    logger.info('')
-    if output_dir:
-        microdf.round(3).to_csv(join(output_dir, 'snr.csv'))
-
-    if mixdb.truth_mutex:
-        macrodf, microdf, wghtdf, mxid_snro = snr_summary(mixdb=mixdb,
-                                                          mixid=':',
-                                                          truth_f=truth[:, 0:-1],
-                                                          predict=predict[:, 0:-1],
-                                                          segsnr=segsnr if compute_segsnr else None,
-                                                          predict_thr=predict_thr)
-
-        logger.info(f'Metrics micro-avg without "Other" class per SNR over all {num_mixtures} mixtures:')
-        logger.info(microdf.round(3).to_string())
-        logger.info('')
-        if output_dir:
-            microdf.round(3).to_csv(join(output_dir, 'snrwo.csv'))
-
-    for snri in snrlist:
-        mxids = mxid_snro[snri]
-        classdf = class_summary(mixdb, mxids, truth, predict, predict_thr)
-        logger.info(f'Metrics per class for SNR {snri} over {len(mxids)} mixtures:')
-        logger.info(classdf.round(3).to_string())
-        logger.info('')
-        if output_dir:
-            classdf.round(3).to_csv(join(output_dir, f'class_snr{snri}.csv'))
-
-
-def main() -> None:
-    from datetime import datetime
-    from os import mkdir
-    from os.path import join
-
-    import h5py
-    from docopt import docopt
-
-    import sonusai
-    from sonusai import SonusAIError
-    from sonusai import create_file_handler
-    from sonusai.utils import read_predict_data
-    from sonusai.utils import trim_docstring
-
-    args = docopt(trim_docstring(__doc__), version=sonusai.__version__, options_first=True)
-
-    verbose = args['--verbose']
-    feature_name = args['--feature']
-    predict_name = args['--predict']
-    predict_threshold = np.array(float(args['--thr']), dtype=np.float32)
-    location = args['LOC']
-
-    mixdb = MixtureDatabase(location)
-
-    # create output directory
-    output_dir = f'evaluate-{datetime.now():%Y%m%d}'
-    try:
-        mkdir(output_dir)
-    except OSError as _:
-        output_dir = f'evaluate-{datetime.now():%Y%m%d-%H%M%S}'
-        try:
-            mkdir(output_dir)
-        except OSError as error:
-            raise SonusAIError(f'Could not create directory, {output_dir}: {error}')
-
-    create_file_handler(join(output_dir, 'evaluate.log'))
-
-    with h5py.File(feature_name, 'r') as f:
-        truth_f = np.array(f['truth_f'])
-        segsnr = np.array(f['segsnr'])
-
-    predict = read_predict_data(predict_name)
-
-    evaluate(mixdb=mixdb,
-             truth=truth_f,
-             segsnr=segsnr,
-             output_dir=output_dir,
-             predict=predict,
-             predict_thr=predict_threshold,
-             verbose=verbose)
-
-    logger.info(f'Wrote results to {output_dir}')
-
-
-if __name__ == '__main__':
-    try:
-        main()
-    except KeyboardInterrupt:
-        logger.info('Canceled due to keyboard interrupt')
-        raise SystemExit(0)