Optionally use individual stft options each stem

jurihock · Mar 25, 2024 · d863f14 · d863f14
1 parent c02f67c
commit d863f14
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/remucs/synthesis.py b/src/remucs/synthesis.py
@@ -87,16 +87,16 @@ def synthesize(file, data, opts):
         stems       = [STEMS.index(stem) for stem in ['bass', 'other', 'vocals']]
         factors     = [pitch] * len(stems)
         quefrencies = [0, 0, opts.quefrency]
-        framesize   = opts.framesize
-        hopsize     = opts.hopsize
+        framesizes  = [opts.framesize] * len(stems)
+        hopsizes    = [opts.hopsize] * len(stems)
 
         for i, stem in enumerate(stems):
             x[stem] = shiftpitch(x[stem],
                 samplerate=sr,
                 factor=factors[i],
                 quefrency=quefrencies[i],
-                framesize=framesize,
-                hopsize=hopsize)
+                framesize=framesizes[i],
+                hopsize=hopsizes[i])
 
     if not opts.quiet:
         if mono: