Skip to content
Extraits de code Groupes Projets
Valider 60a649a9 rédigé par Sting's avatar Sting
Parcourir les fichiers

Use Madmom's CNN implementation

parent 57726b35
Aucune branche associée trouvée
Aucune étiquette associée trouvée
1 requête de fusion!2CNN + Spectral segmentation
*
!.gitignore
!README.md
!requirements.txt
!extractWav.sh
!extractAss.sh
!karaUtils.py
!autokara.py
!assUtils.py
!process_train_data.sh
!cnn_prepare_data.py
!cnn_train.py
!*/cnn/segment.py
!*/cnn/music_processor.py
!*/cnn/model.py
!rosa/*.py
media/
\ No newline at end of file
__pycache__/
data/
env/
media/
models/
......@@ -6,7 +6,7 @@ import shlex
from pathlib import Path
from assUtils import AssWriter
from cnn.segment import segment
from cnn_madmom.segment import segment
parser = argparse.ArgumentParser(description='AutoKara - Automatic karaoke timing tool')
......@@ -42,7 +42,7 @@ else:
print("Identifying syl starts...")
onsets = segment(vocals_file)
syls = [[t, ''] for t in onsets]
syls = [[t, 'la'] for t in onsets]
print("Syls found, writing ASS file...")
writer = AssWriter()
......
import madmom
import numpy as np
import sys
import matplotlib.pyplot as plt
from scipy.ndimage.filters import maximum_filter
def segment(songfile):
cnn = madmom.features.onsets.CNNOnsetProcessor()
proc = madmom.features.onsets.OnsetPeakPickingProcessor(threshold=0.4, smooth=5, delay=0, fps=100)
activation_function = cnn(songfile, num_channels=1)
#onsets = proc(activation_function)
onsets = madmom.features.onsets.peak_picking(activation_function, 0.4, smooth=10)/100
print(onsets)
return onsets
\ No newline at end of file
0% Chargement en cours ou .
You are about to add 0 people to the discussion. Proceed with caution.
Terminez d'abord l'édition de ce message.
Veuillez vous inscrire ou vous pour commenter