From beaf53c7e0afaf9327e2f72e87c590a56310e78d Mon Sep 17 00:00:00 2001 From: marcsello Date: Wed, 16 Sep 2020 19:44:20 +0200 Subject: [PATCH] Fixed incorrectly scanning directories --- cnn_classification_service/cnn_classifier.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cnn_classification_service/cnn_classifier.py b/cnn_classification_service/cnn_classifier.py index df8570b..9e4c194 100644 --- a/cnn_classification_service/cnn_classifier.py +++ b/cnn_classification_service/cnn_classifier.py @@ -39,9 +39,10 @@ class Classifier(object): librosa.display.specshow(librosa.power_to_db(spectogram, ref=numpy.max)) target_dir = tempfile.mkdtemp() + wav_basename = os.path.basename(wav_filename) # Change extension to jpg... mert 110% biztos vagyok benne hogy a keras nem bírná beolvasni máshogy - file_name = os.path.join(target_dir, "unknown", f"{wav_filename[:-4]}.jpg") + file_name = os.path.join(target_dir, "unknown", f"{wav_basename[:-4]}.jpg") matplotlib.pyplot.savefig(file_name, dpi=400, bbox_inches='tight', pad_inches=0) matplotlib.pyplot.close()