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

Moved frequently changed files to /dev/shm

Browse Source
This commit is contained in:
Pünkösd Marcell
2021-07-28 15:40:51 +02:00
parent 439a3faed6
commit ee6c9d7b47
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cnn_classification_service/cnn_classifier.py
View File

@@ -38,7 +38,7 @@ class Classifier(object):
spectogram = librosa.feature.melspectrogram(y=clip, sr=sample_rate)
librosa.display.specshow(librosa.power_to_db(spectogram, ref=numpy.max))
target_dir = tempfile.mkdtemp()
target_dir = tempfile.mkdtemp(dir="/dev/shm")
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

2
cnn_classification_service/magic_doer.py
View File

@@ -17,7 +17,7 @@ class MagicDoer:
@classmethod
def run_everything(cls, parameters: dict) -> dict:
tag = parameters['tag']
sample_file_handle, sample_file_path = tempfile.mkstemp(prefix=f"{tag}_", suffix=".wav")
sample_file_handle, sample_file_path = tempfile.mkstemp(prefix=f"{tag}_", suffix=".wav", dir="/dev/shm")
response = None
try: