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

sonusai/utils/__init__.py

@@ -9,6 +9,8 @@ from .audio_devices import get_input_devices
 from .braced_glob import braced_glob
 from .braced_glob import braced_iglob
 from .calculate_input_shape import calculate_input_shape
+from .compress import power_compress
+from .compress import power_uncompress
 from .convert_string_to_number import convert_string_to_number
 from .create_timestamp import create_timestamp
 from .create_ts_name import create_ts_name

sonusai/utils/compress.py

@@ -0,0 +1,25 @@
+from sonusai.mixture import AudioF
+
+
+def power_compress(feature: AudioF) -> AudioF:
+    import numpy as np
+
+    mag = np.abs(feature)
+    phase = np.angle(feature)
+    mag = mag ** 0.3
+    real_compress = mag * np.cos(phase)
+    imag_compress = mag * np.sin(phase)
+
+    return real_compress + 1j * imag_compress
+
+
+def power_uncompress(feature: AudioF) -> AudioF:
+    import numpy as np
+
+    mag = np.abs(feature)
+    phase = np.angle(feature)
+    mag = mag ** (1. / 0.3)
+    real_uncompress = mag * np.cos(phase)
+    imag_uncompress = mag * np.sin(phase)
+
+    return real_uncompress + 1j * imag_uncompress

sonusai/utils/energy_f.py

@@ -1,5 +1,4 @@
-from pyaaware import ForwardTransform
-
+from sonusai import ForwardTransform
 from sonusai.mixture import AudioF
 from sonusai.mixture import AudioT
 from sonusai.mixture import EnergyF
@@ -19,7 +18,7 @@ def compute_energy_f(frequency_domain: AudioF = None,
     :return: Frequency domain per-bin energy data [frames, bins]
     """
     import numpy as np
-
+    import torch
     from sonusai import SonusAIError
 
     if frequency_domain is None:
@@ -28,7 +27,7 @@ def compute_energy_f(frequency_domain: AudioF = None,
         if transform is None:
             raise SonusAIError('Must provide ForwardTransform object')
 
-        frequency_domain, _ = transform.execute_all(time_domain)
+        frequency_domain = transform.execute_all(torch.from_numpy(time_domain))[0].numpy()
 
     frames, bins = frequency_domain.shape
     result = np.empty((frames, bins), dtype=np.float32)

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

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

@@ -1,4 +1,4 @@
-sonusai/__init__.py,sha256=s77La_fCT4TJtXR8fT0jHrWgGwOQI5qonyS3LStzsyw,2980
+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
@@ -10,14 +10,14 @@ sonusai/doc/__init__.py,sha256=rP5Hgn0Iys_xkuv4caxngdqehuU4zLZsiKuv8Nde67M,19
 sonusai/doc/doc.py,sha256=8gftMlCaVa0hIA_BjSHbEYgSTq_BAZwC5Ipk-BFgGp8,22919
 sonusai/doc.py,sha256=l8CaFgLI8mqx4tn0aXfxKqa2dy9GgC0zjYxZAkpmi1E,878
 sonusai/genft.py,sha256=OzET3iTE-QhrUckzidfZvCDXZlAxIF5Xe5NEf856Vvk,5662
-sonusai/genmetrics.py,sha256=YS5ZuhOR77-wGH8ju7z_ihQSDhYhV_iJWlWWMemLIGU,4998
+sonusai/genmetrics.py,sha256=dyonwsztn23-iPPg_5FOzai-n6veuThsvrdt6iBycTE,5007
 sonusai/genmix.py,sha256=TU5aTebGHsbfwsRbynYbegGBelSma9khuQkDk0dFE3I,7075
 sonusai/genmixdb.py,sha256=gF2qas1tH9MHEFLoEMrN3kYVm-vhAKaOuZ8ev-w4MQM,19553
 sonusai/gentcst.py,sha256=W1ZO3xs7CoZkFcvOTH-FLJOIA4I7Wzb0HVRC3hGGSaM,20223
 sonusai/lsdb.py,sha256=oLyGoq46hkravDUXhI-wwhKdX_GESoOD5-E7nEJl7rc,5996
 sonusai/main.py,sha256=GC-pQrSqx9tWwIcmEo6V9SraEv5KskBLS_W_wz-f2ZM,2509
-sonusai/metrics/__init__.py,sha256=GEenHTytN-_qMAF7UJypqkArudJsHyctGSKd0bKUniQ,807
-sonusai/metrics/calc_audio_stats.py,sha256=u-HIyrqJUCb0G7WR865TcpTVkT2lsCnb2rwB67IgzEc,1308
+sonusai/metrics/__init__.py,sha256=uLRKJNPxrSYtLL5zhPyr8mGGVdterS_Pi9kYDru1GHA,858
+sonusai/metrics/calc_audio_stats.py,sha256=R_s-j4GA_QuFBMjR-tZn3v7QkRiIPE0QlkmczCyt2aw,1553
 sonusai/metrics/calc_class_weights.py,sha256=dyY7daEIf5Ms5tfTf6wF0fkx_GnMADHOZR_rtsfGoVM,3933
 sonusai/metrics/calc_optimal_thresholds.py,sha256=9fRfwl-aKAbzHJyqGHv4o8BpZXG9HHB7zUJObHXfYM4,3522
 sonusai/metrics/calc_pcm.py,sha256=n_QoDwdM09HZx0zMbCkJR07e9fCLo03Ilpi1nqdizRg,1906
@@ -25,30 +25,30 @@ sonusai/metrics/calc_pesq.py,sha256=vO3HnQfyzYLy6VILhP5E-Yi6HkI6T9ob_CEfPo6NG50,
 sonusai/metrics/calc_phase_distance.py,sha256=Oh7UrH4ekm48pIhOnAOmkSr1OWueq2GUlwJheoAQmnk,1905
 sonusai/metrics/calc_sa_sdr.py,sha256=pup1uzHbSqV3JprYCXmu_hbWILqRME5KAnFJGe4WXCY,2566
 sonusai/metrics/calc_sample_weights.py,sha256=0O2EH1-FKlCa0HFgKPUF1BJiknR1hCH7zLbXnoXH7Ag,945
-sonusai/metrics/calc_snr_f.py,sha256=cLRUYeyRuoDtXWOgrgSzFP3EMOBlRaBcPM1BcxpEU_M,1199
-sonusai/metrics/calc_speech.py,sha256=i-BjYiSY-M3YSbKFUOW4m7m4frkF4ikyJgyizhDjGm0,14448
+sonusai/metrics/calc_segsnr_f.py,sha256=9-fFrQ2Ilcayab8bJw_nBqyF4Q15Nqyef_EFihxaMRs,3125
+sonusai/metrics/calc_speech.py,sha256=r0KKY-wlNN9mEkIRUhh1hap73A4rw4G9QzSOChXQorc,14437
 sonusai/metrics/calc_wer.py,sha256=7VVW_YsbeysKOZOTtvXVAvWi8lb0PZtu-Xg2HiXomWc,2483
 sonusai/metrics/calc_wsdr.py,sha256=9wiCJsGgRGHzBpHv0zSxorMsN_JZNBPyge280Hd1W0w,2531
 sonusai/metrics/class_summary.py,sha256=4Mb25nuk6eqotnQSFMuOQL3zofGcpNXDfDlPa513ZLQ,3286
 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=5M8wEFLkySuaFUKFLhXGPs9SNt2fLwsr6QqEYi7-eDY,5358
+sonusai/mixture/__init__.py,sha256=uX0NT0EWLYK0rR2jgZiMR0VQ7wM6lV6FU2RLzWn485w,5466
 sonusai/mixture/audio.py,sha256=2u7nnKdKwXRA6YWVwCurst5D6CQinKZj5jlm9LhXe9c,2445
 sonusai/mixture/augmentation.py,sha256=Blb90tdTwBOj5w9tRcYyS5H67YJuFiXsGqwZWd7ON4g,10468
 sonusai/mixture/class_count.py,sha256=_wFnVl2yEOnbor7pLg7cYOUeX6nioov-03Cv3SEbh2k,996
 sonusai/mixture/config.py,sha256=Fu6-HkenI_h7BdOTuGX9B-vsbLJD5pvTPd_LAncZgI4,22828
 sonusai/mixture/constants.py,sha256=90qaRIEcmIoS3Od5h_UP0_SkkvG2aE_eYPv6WsIktC0,1427
-sonusai/mixture/datatypes.py,sha256=uYNToIngMB4m0pgmaAFj7H27FvrMCiwaAAXDXkvIB6w,9100
+sonusai/mixture/datatypes.py,sha256=2vegllgZcmFLq5NjqS7Lo97dOpOJOAj0Eml4ggP_tGo,10966
 sonusai/mixture/db_datatypes.py,sha256=GDYbcSrlgUJsesiUUNnR4s5aBkMgviiNSQDaBcgYX7I,1428
 sonusai/mixture/eq_rule_is_valid.py,sha256=MpQwRA5M76wSiQWEI1lW2cLFdPaMttBLcQp3tWD8efM,1243
-sonusai/mixture/feature.py,sha256=Rwuf82IoXzhHPGbKYVGcatImF_ssBf_FfvbqghVPXtg,4116
+sonusai/mixture/feature.py,sha256=bHAPRaYGyS-ZTOb-RLCwDau7n1NDKsVEW30Gd9SRZYo,3676
 sonusai/mixture/generation.py,sha256=W3n6ipI-dxg4Wj6YBJn8RTpFqkAyIXzxwObeFbSLq08,42801
-sonusai/mixture/helpers.py,sha256=eC9ZysEa-83VLKen_9PKWzr8w9dkHj4lp6rMB2fNLbg,24759
+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=WPXuziS0SePtKkd7qtkE7Db8rV-TKQoUBvs4GPieTek,57332
+sonusai/mixture/mixdb.py,sha256=mr9Ck4p_mCfvz1PXoUgjWcw9F-Rlw3uGiDizUvPqo2A,64359
 sonusai/mixture/soundfile_audio.py,sha256=BwO4lftNvrhoPTJERONcrpxSpM2fjO6kL_e5Ylz742A,4220
-sonusai/mixture/sox_audio.py,sha256=Pu5RLE8lKKlGOzZsbfcVzbDvy7YPGBuGpbD_PxowFqA,12511
+sonusai/mixture/sox_audio.py,sha256=DbHuyLtEuQYtKsIRxx6g1webW_LsdgLz52P5VO37MqI,17119
 sonusai/mixture/sox_augmentation.py,sha256=kBWPrsFk0EBi71nLcKt5v0GA34bY7g9D9x0cEamNWbU,4564
 sonusai/mixture/spectral_mask.py,sha256=8AkCwhy-PSdP1Uri9miKZP-bXFYnFcH_c9xZCGrHavU,2071
 sonusai/mixture/target_class_balancing.py,sha256=NTNiKZH0_PWLooeow0l41CjJKK8ZTMVbUqz9ZkaNtWk,4900
@@ -59,16 +59,15 @@ 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=okFJeOf43NxfdLqWFCBA2pOGqujRlNDYdAcwwR_m8z8,2875
-sonusai/mixture/truth_functions/energy.py,sha256=xWxjrHM1xPitRFML-X7ruqXMmXVII5MKDjU38WO6bG0,4863
+sonusai/mixture/truth_functions/data.py,sha256=UcjldGSTQIQ5hq38zVKHx7X6GMmXcsChU2lyvQb8vpI,2752
+sonusai/mixture/truth_functions/energy.py,sha256=BSdD1sHFs4bDnB09d5k21yb-2qECF4PScIAejKrq5uU,4907
 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=gy2OGAV_uIrZunNBRC6kZx3qg48TSzNI_HaXGzPKg3s,5763
+sonusai/mixture/truth_functions/sed.py,sha256=ak6_GAdw2VxG2svHvKWHLvkL5nCDon0rD2osJbzkU3E,2512
+sonusai/mixture/truth_functions/target.py,sha256=XypzXVMi24Ys13TiEM9JFY_cvHK61Lop7AB-UL5AObw,5731
 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=pHaJZtms7aj4r6sgqQnEGVi6Gg8H_V29szigogV1vZ8,5002
 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
@@ -82,7 +81,7 @@ sonusai/speech/vctk.py,sha256=EAMEBAzjZUI6dw15n-yI2oCN-H4tzM9t4aUVlOxpAbo,1540
 sonusai/speech/voxceleb.py,sha256=aJGN0yDb2LFLmCKmRzmUEjpZWQ-QGWw6XWOpy9967AI,2686
 sonusai/summarize_metric_spenh.py,sha256=OiZe_bhCq5esXNhsOkHDD7g4ssYrpENDHvDVoPzV9iw,1822
 sonusai/tplot.py,sha256=85T6OPZfxVegHBiSuilFpdgCNMEE0VKAuciNy4rCY5Y,14544
-sonusai/utils/__init__.py,sha256=UEM7SkO_xxu0VW-u_zI3e8Xih2CX4aqYjqr61LO5EQA,2271
+sonusai/utils/__init__.py,sha256=ivPBOtc6m5xnHC1BKVUNxcXOxwaPc-u3jgS4jLKoGio,2347
 sonusai/utils/asl_p56.py,sha256=-bvQpd-jRQVURbkZJpRoyEAq6gTv9Rc3oFDbh5_lcjY,3861
 sonusai/utils/asr.py,sha256=V0-cW56KS6aOSakSVwQIbhkqEkKf5l8T11b8ciNxWDw,2132
 sonusai/utils/asr_functions/__init__.py,sha256=JyHK67s97bw7QzrlkboWhws4yNytdPatqzLJxfwx-yw,43
@@ -90,13 +89,14 @@ sonusai/utils/asr_functions/aaware_whisper.py,sha256=NPqc62wKhshYecu368Yvl2xcImp
 sonusai/utils/audio_devices.py,sha256=LgaXTln1oRArBzaet3rZiIO2plgtaThuGBc3sJ_sLlo,1414
 sonusai/utils/braced_glob.py,sha256=Z_XIpPK17QiP1JbzAnUC5w3oyG8ZovoyM22Wh-Q_vWU,1675
 sonusai/utils/calculate_input_shape.py,sha256=63ILxibYKuTQozY83QN8Y2OOhBEbW_1X47Q0askcHDM,984
+sonusai/utils/compress.py,sha256=NSiYXSnA2WifFYbq7ZSv0wX1YKwcPNwAkrWc4-9rbDM,606
 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
 sonusai/utils/docstring.py,sha256=JBecAq_a7KSzZ04tan0BlFA9SzhOQqY9A7Ue85kFQdU,1446
-sonusai/utils/energy_f.py,sha256=AsoGo3TCMnj9Kzqb9HUAYOL6vPGUseu1p-8z4LG-QYo,1462
+sonusai/utils/energy_f.py,sha256=EALYk5C7vpvh8v417lneM1t4gZTfVyIlPVcF9e6-m_U,1502
 sonusai/utils/engineering_number.py,sha256=6lWRg-XW3FEllh-Zui1WHYcDnsBEY9cqcwcXPIs2-6o,5515
 sonusai/utils/get_frames_per_batch.py,sha256=xnq4tV7MT74N0H6b5ZsiAezqdXucboCLQw1Np9XpZbs,134
 sonusai/utils/get_label_names.py,sha256=bfFV_iFbXdtDp70zAz_CQQN9-QxBw_kt8vyAcr3TITM,828
@@ -119,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.6.dist-info/METADATA,sha256=L4wl__zTyqiH1SEMDPIIJcgLM7Uxqa6lQ9EMXVjXRlI,2591
-sonusai-0.18.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-sonusai-0.18.6.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
-sonusai-0.18.6.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/metrics/calc_snr_f.py

@@ -1,34 +0,0 @@
-import numpy as np
-
-from sonusai.mixture.datatypes import Segsnr
-from sonusai.mixture.datatypes import SnrFMetrics
-
-
-def calc_snr_f(segsnr_f: Segsnr) -> SnrFMetrics:
-    """Calculate metrics of snr_f truth data.
-
-    For now, includes mean and variance of the raw values (usually energy)
-    and mean and standard deviation of the dB values (10 * log10).
-    """
-    if np.count_nonzero(segsnr_f) == 0:
-        # If all entries are zeros
-        return SnrFMetrics(0, 0, -np.inf, 0)
-
-    tmp = np.ma.array(segsnr_f, mask=np.logical_not(np.isfinite(segsnr_f)), dtype=np.float32)
-    if np.ma.count_masked(tmp) == np.ma.size(tmp, axis=0):
-        # If all entries are infinite
-        return SnrFMetrics(np.inf, 0, np.inf, 0)
-
-    snr_mean = np.mean(tmp, axis=0)
-    snr_var = np.var(tmp, axis=0)
-
-    tmp = 10 * np.ma.log10(tmp)
-    if np.ma.count_masked(tmp) == np.ma.size(tmp, axis=0):
-        # If all entries are masked, special case where all inputs are either 0 or infinite
-        snr_db_mean = -np.inf
-        snr_db_std = np.inf
-    else:
-        snr_db_mean = np.mean(tmp, axis=0)
-        snr_db_std = np.std(tmp, axis=0)
-
-    return SnrFMetrics(snr_mean, snr_var, snr_db_mean, snr_db_std)

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 pyaaware import AawareInverseTransform
-
-    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=AawareInverseTransform(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()