Remplacement modèle classification par genre

This commit is contained in:
2025-12-22 14:29:05 +01:00
parent 90c841310c
commit d67d5ecadd
3 changed files with 29 additions and 21 deletions

View File

@@ -47,9 +47,15 @@ download_model "discogs-effnet-bs64-1.pb" \
# Download classification heads
echo ""
echo "Downloading classification heads..."
download_model "mtg_jamendo_genre-discogs-effnet-1.pb" \
"$CLASS_HEADS_URL/mtg_jamendo_genre/mtg_jamendo_genre-discogs-effnet-1.pb"
# Genre: Discogs400 (professional taxonomy with 400 genres)
download_model "genre_discogs400-discogs-effnet-1.pb" \
"$CLASS_HEADS_URL/genre_discogs400/genre_discogs400-discogs-effnet-1.pb"
download_model "genre_discogs400-discogs-effnet-1.json" \
"$CLASS_HEADS_URL/genre_discogs400/genre_discogs400-discogs-effnet-1.json"
# Mood & Instrument: MTG-Jamendo
download_model "mtg_jamendo_moodtheme-discogs-effnet-1.pb" \
"$CLASS_HEADS_URL/mtg_jamendo_moodtheme/mtg_jamendo_moodtheme-discogs-effnet-1.pb"