https://github.com/cschen1205/java-deep-learning-audio
Audio Deep Learning Project in Java
https://github.com/cschen1205/java-deep-learning-audio
audio-classification audio-embedding audio-processing audio-search genres-classification music-classification music-recommendation search-engine signal-processing
Last synced: 6 months ago
JSON representation
Audio Deep Learning Project in Java
- Host: GitHub
- URL: https://github.com/cschen1205/java-deep-learning-audio
- Owner: cschen1205
- License: mit
- Created: 2018-04-02T03:41:07.000Z (over 7 years ago)
- Default Branch: master
- Last Pushed: 2018-04-02T05:43:14.000Z (over 7 years ago)
- Last Synced: 2025-03-24T12:17:31.349Z (7 months ago)
- Topics: audio-classification, audio-embedding, audio-processing, audio-search, genres-classification, music-classification, music-recommendation, search-engine, signal-processing
- Language: Java
- Size: 96.1 MB
- Stars: 3
- Watchers: 2
- Forks: 2
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# java-deep-learning-audio
Audio Deep Learning Project in Java
# Predict Music Genres
The [sample codes](src/main/java/com/github/cschen1205/tensorflow/DeepAudioDemo.java) shows how to
[DeepAudio](src/main/java/com/github/cschen1205/tensorflow/DeepAudioTensorflow.java) to predict the genres of an
music file:
```java
import com.github.cschen1205.tensorflow.commons.FileUtils;
import com.github.cschen1205.tensorflow.search.models.AudioSearchEntry;
import java.io.File;
import java.util.Collections;
import java.util.List;
public class MusicGenrePredictionDemo {
public static void main(String[] args){
DeepAudio classifier = new DeepAudioTensorflow();
String folderStoringMusicFiles = "music_samples";
List paths = FileUtils.getAudioFilePaths(folderStoringMusicFiles, ".au");
Collections.shuffle(paths);
for (String path : paths) {
System.out.println("Predicting " + path + " ...");
File f = new File(path);
String label = classifier.predictMusicGenres(f);
System.out.println("Predicted: " + label);
}
}
}
```
# Music Search
The [sample codes](src/main/java/com/github/cschen1205/tensorflow/DeepAudioDemo.java) shows how to
[DeepAudio](src/main/java/com/github/cschen1205/tensorflow/DeepAudioTensorflow.java) to search for similar musics stored
in your local folder using search query which is the music file of interest:
```java
import com.github.cschen1205.tensorflow.commons.FileUtils;
import com.github.cschen1205.tensorflow.search.models.AudioSearchEntry;
import java.io.File;
import java.util.Collections;
import java.util.List;
public class MusicSearchEngineDemo {
public static void main(String[] args){
DeepAudio searchEngine = new DeepAudioTensorflow();
if(!searchEngine.loadMusicIndexDbIfExists()) {
String folderStoringMusicFiles = "music_samples";
searchEngine.indexMusicFiles(FileUtils.getAudioFiles(folderStoringMusicFiles, ".au"));
searchEngine.saveMusicIndexDb();
}
int pageIndex = 0;
int pageSize = 20;
boolean skipPerfectMatch = true;
File sample_file = new File("mp3_samples/example.mp3");
System.out.println("querying similar music to " + sample_file.getName());
List result = searchEngine.query(sample_file, pageIndex, pageSize, skipPerfectMatch);
for(int i=0; i < result.size(); ++i){
System.out.println("# " + i + ": " + result.get(i).getPath() + " (distSq: " + result.get(i).getDistance() + ")");
}
}
}
```
# Song Recommend-er based on user history
The [sample codes](src/main/java/com/github/cschen1205/tensorflow/DeepAudioDemo.java) shows how to
[DeepAudio](src/main/java/com/github/cschen1205/tensorflow/DeepAudioTensorflow.java) to recommends other songs based
on user's recent listening history:
```java
import com.github.cschen1205.tensorflow.commons.FileUtils;
import com.github.cschen1205.tensorflow.search.models.AudioSearchEntry;
import java.io.File;
import java.util.Collections;
import java.util.List;
public class SongRecommendationDemo {
private static UserMusicHistory getUserMusicHistory() {
UserMusicHistory userHistory = new UserMusicHistory();
List audioFiles = FileUtils.getAudioFilePaths("music_samples", ".au");
Collections.shuffle(audioFiles);
for(int i=0; i < 40; ++i){
String filePath = audioFiles.get(i);
userHistory.logAudio(filePath);
try {
Thread.sleep(100L);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
return userHistory;
}
public static void main(String[] args){
UserMusicHistory userHistory = getUserMusicHistory();
DeepAudio recommender = new DeepAudioTensorflow();
if(!recommender.loadMusicIndexDbIfExists()) {
recommender.indexMusicFiles(FileUtils.getAudioFiles("music_samples", ".au"));
recommender.saveMusicIndexDb();
}
System.out.println(userHistory.head(10));
int k = 10;
List result = recommender.recommends(userHistory.getHistory(), k);
for(int i=0; i < result.size(); ++i){
AudioSearchEntry entry = result.get(i);
System.out.println("Search Result #" + (i+1) + ": " + entry.getPath());
}
}
}
```