fedir/caveman
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update cavemanml.py

Browse Source
This commit is contained in:
fedir
2026-01-07 12:37:36 +01:00
parent 7633a21e1d
commit 79f93e5c29
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cavemanml.py
View File

@@ -66,12 +66,12 @@ def main():
model.eval() model.eval()
enhance_audio( enhance_audio(
model, model,
"./mirror_mirror_compressed_64.mp3", "./examples/mirror_mirror/mirror_mirror_compressed_64.mp3",
"./mirror_mirror_decompressed_64_mse.wav", "./examples/mirror_mirror/mirror_mirror_decompressed_64_mse.wav",
) )
# Load # Load
x, sr = librosa.load("./mirror_mirror_compressed_64.mp3", sr=SR) x, sr = librosa.load("./examples/mirror_mirror/mirror_mirror_compressed_64.mp3", sr=SR)
# Convert to log-mag # Convert to log-mag
X = audio_to_logmag(x) # (513, T) X = audio_to_logmag(x) # (513, T)
@@ -90,7 +90,7 @@ def main():
import soundfile as sf import soundfile as sf
# Save # Save
sf.write("./mirror_mirror_compressed_64_STFT.mp3", y_reconstructed, sr) sf.write("./examples/mirror_mirror/mirror_mirror_compressed_64_STFT.mp3", y_reconstructed, sr)
return return
# Data # Data