birbnetes/cnn-classification-service
4
0
Fork 0
Code Pull Requests Releases Wiki Activity

Fixed incorrectly scanning directories
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Pünkösd Marcell
2020-09-16 19:44:20 +02:00
parent b3770a6de9
commit beaf53c7e0
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
cnn_classification_service/cnn_classifier.py
View File

@@ -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()