Musicnn: Only load the first 120s to avoid OOM

nextcloud · Oct 24, 2021 · b29da0a · b29da0a
1 parent 4a5be7a
commit b29da0a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/classifier_musicnn.js b/src/classifier_musicnn.js
@@ -49,6 +49,9 @@ function findRule(className) {
 
 if (process.argv.length < 3) throw new Error('Incorrect arguments: node classify.js ...<IMAGE_FILES> | node classify.js -')
 
+/**
+ *
+ */
 async function main() {
 	const modelPath = path.resolve(__dirname, '..', 'music_model')
 

diff --git a/src/musicnn/MusicnnModel.js b/src/musicnn/MusicnnModel.js
@@ -55,6 +55,7 @@ class MusicnnModel {
 			'-ac', '1',
 			'-ar', '8000',
 			'-acodec', 'pcm_s16le',
+			'-t', '120',
 			'-',
 		], { encoding: null, stripFinalNewline: false })
-Original file line number
+Diff line change
@@ Expand Up / @@ -49,6 +49,9 @@ function findRule(className) { @@
     if (process.argv.length < 3) throw new Error('Incorrect arguments: node classify.js ...<IMAGE_FILES> | node classify.js -')
+    /**
+     *
+     */
     async function main() {
     	const modelPath = path.resolve(__dirname, '..', 'music_model')
@@ Expand Down @@