Spaces:
Running
Running
update fx norm
Browse files- __pycache__/config.cpython-311.pyc +0 -0
- __pycache__/inference.cpython-311.pyc +0 -0
- config.py +1 -0
- inference.py +2 -2
- models/fxnorm_feat.npy +3 -0
- modules/__pycache__/common_audioeffects.cpython-311.pyc +0 -0
- modules/__pycache__/common_miscellaneous.cpython-311.pyc +0 -0
- modules/__pycache__/data_normalization.cpython-311.pyc +0 -0
- modules/__pycache__/fx_utils.cpython-311.pyc +0 -0
- modules/__pycache__/loss.cpython-311.pyc +0 -0
- modules/__pycache__/normalization_imager.cpython-311.pyc +0 -0
- modules/__pycache__/utils_data_normalization.cpython-311.pyc +0 -0
- requirements.txt +2 -1
__pycache__/config.cpython-311.pyc
CHANGED
Binary files a/__pycache__/config.cpython-311.pyc and b/__pycache__/config.cpython-311.pyc differ
|
|
__pycache__/inference.cpython-311.pyc
CHANGED
Binary files a/__pycache__/inference.cpython-311.pyc and b/__pycache__/inference.cpython-311.pyc differ
|
|
config.py
CHANGED
@@ -11,6 +11,7 @@ parser.add_argument("--num_steps", type=int, default=100, help="Number of optimi
|
|
11 |
parser.add_argument("--af_weights", nargs='+', type=float, default=[0.1, 0.001, 1.0, 1.0, 0.1], help="Weights for AudioFeatureLoss")
|
12 |
parser.add_argument("--sample_rate", type=int, default=44100, help="Sample rate for AudioFeatureLoss")
|
13 |
parser.add_argument("--path_to_config", type=str, default='networks/configs.yaml', help="Path to network architecture configuration file")
|
|
|
14 |
|
15 |
args = parser.parse_args()
|
16 |
|
|
|
11 |
parser.add_argument("--af_weights", nargs='+', type=float, default=[0.1, 0.001, 1.0, 1.0, 0.1], help="Weights for AudioFeatureLoss")
|
12 |
parser.add_argument("--sample_rate", type=int, default=44100, help="Sample rate for AudioFeatureLoss")
|
13 |
parser.add_argument("--path_to_config", type=str, default='networks/configs.yaml', help="Path to network architecture configuration file")
|
14 |
+
parser.add_argument("--fx_norm_feature_path", type=str, default='models/fxnorm_feat.npy', help="Path to precomputed fx normalization features")
|
15 |
|
16 |
args = parser.parse_args()
|
17 |
|
inference.py
CHANGED
@@ -11,6 +11,7 @@ currentdir = os.path.dirname(os.path.realpath(__file__))
|
|
11 |
sys.path.append(os.path.dirname(currentdir))
|
12 |
from networks import Dasp_Mastering_Style_Transfer, Effects_Encoder
|
13 |
from modules.loss import AudioFeatureLoss, Loss
|
|
|
14 |
|
15 |
|
16 |
def convert_audio(wav: torch.Tensor, from_rate: float,
|
@@ -32,8 +33,7 @@ class MasteringStyleTransfer:
|
|
32 |
|
33 |
self.fx_normalizer = Audio_Effects_Normalizer(precomputed_feature_path=args.fx_norm_feature_path, \
|
34 |
STEMS=['mixture'], \
|
35 |
-
EFFECTS=['eq', 'imager', 'loudness']
|
36 |
-
audio_extension=args.audio_extension)
|
37 |
|
38 |
def load_effects_encoder(self):
|
39 |
effects_encoder = Effects_Encoder(self.args.cfg_enc)
|
|
|
11 |
sys.path.append(os.path.dirname(currentdir))
|
12 |
from networks import Dasp_Mastering_Style_Transfer, Effects_Encoder
|
13 |
from modules.loss import AudioFeatureLoss, Loss
|
14 |
+
from modules.data_normalization import Audio_Effects_Normalizer
|
15 |
|
16 |
|
17 |
def convert_audio(wav: torch.Tensor, from_rate: float,
|
|
|
33 |
|
34 |
self.fx_normalizer = Audio_Effects_Normalizer(precomputed_feature_path=args.fx_norm_feature_path, \
|
35 |
STEMS=['mixture'], \
|
36 |
+
EFFECTS=['eq', 'imager', 'loudness'])
|
|
|
37 |
|
38 |
def load_effects_encoder(self):
|
39 |
effects_encoder = Effects_Encoder(self.args.cfg_enc)
|
models/fxnorm_feat.npy
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:37d432620d81d544407615d2a1f3d92756e424b7107f661cb84ee0f9ff430007
|
3 |
+
size 131649
|
modules/__pycache__/common_audioeffects.cpython-311.pyc
ADDED
Binary file (76.8 kB). View file
|
|
modules/__pycache__/common_miscellaneous.cpython-311.pyc
ADDED
Binary file (10 kB). View file
|
|
modules/__pycache__/data_normalization.cpython-311.pyc
ADDED
Binary file (16.9 kB). View file
|
|
modules/__pycache__/fx_utils.cpython-311.pyc
ADDED
Binary file (16.7 kB). View file
|
|
modules/__pycache__/loss.cpython-311.pyc
CHANGED
Binary files a/modules/__pycache__/loss.cpython-311.pyc and b/modules/__pycache__/loss.cpython-311.pyc differ
|
|
modules/__pycache__/normalization_imager.cpython-311.pyc
ADDED
Binary file (4.74 kB). View file
|
|
modules/__pycache__/utils_data_normalization.cpython-311.pyc
ADDED
Binary file (48.2 kB). View file
|
|
requirements.txt
CHANGED
@@ -12,4 +12,5 @@ torchcomp==0.1.3
|
|
12 |
pytorch-lightning==2.4.0
|
13 |
julius==0.2.7
|
14 |
pyloudnorm==0.1.1
|
15 |
-
matplotlib==3.8.4
|
|
|
|
12 |
pytorch-lightning==2.4.0
|
13 |
julius==0.2.7
|
14 |
pyloudnorm==0.1.1
|
15 |
+
matplotlib==3.8.4
|
16 |
+
git+https://github.com/csteinmetz1/pymixconsole
|