jhtonyKoo commited on
Commit
0247649
1 Parent(s): 6bbce1b

update fx norm

Browse files
__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