Introduce numpy and torch transforms (#1705)

* Refactor audio processing functions * Add tests for numpy transforms * Fix imports * Fix imports2
coqui-ai · Aug 8, 2022 · d46fbc2 · d46fbc2
1 parent 7fd9b89
commit d46fbc2
Show file tree

Hide file tree

Showing 10 changed files with 701 additions and 172 deletions.
diff --git a/TTS/encoder/models/resnet.py b/TTS/encoder/models/resnet.py
@@ -1,7 +1,7 @@
 import torch
 from torch import nn
 
-# from TTS.utils.audio import TorchSTFT
+# from TTS.utils.audio.torch_transforms import TorchSTFT
 from TTS.encoder.models.base_encoder import BaseEncoder
 
 

diff --git a/TTS/tts/layers/losses.py b/TTS/tts/layers/losses.py
@@ -8,7 +8,7 @@
 
 from TTS.tts.utils.helpers import sequence_mask
 from TTS.tts.utils.ssim import SSIMLoss as _SSIMLoss
-from TTS.utils.audio import TorchSTFT
+from TTS.utils.audio.torch_transforms import TorchSTFT
 
 
 # pylint: disable=abstract-method

diff --git a/TTS/utils/audio/__init__.py b/TTS/utils/audio/__init__.py
@@ -0,0 +1 @@
+from TTS.utils.audio.processor import AudioProcessor
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		from TTS.utils.audio.processor import AudioProcessor