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

Added more stuff to results

Browse Source
This commit is contained in:
Pünkösd Marcell
2021-06-14 03:12:44 +02:00
parent a64b14b06f
commit 07689594b3
3 changed files with 27 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
cnn_classification_service/cnn_classifier.py
View File

@@ -53,7 +53,7 @@ class Classifier(object):
return target_dir, file_name # Az unknown nélkülivel kell visszatérni
def _run_predictor(self, directory: str) -> list:
def _run_predictor(self, directory: str) -> Tuple[str, dict]:
predict_generator = self.datagen.flow_from_directory(
directory=directory,
batch_size=128,
@@ -77,11 +77,13 @@ class Classifier(object):
}
labels = dict((v, k) for k, v in labels.items())
predictions = [labels[k] for k in predicted_class_indices]
labeled_predictions = {labels[i]: p for i, p in enumerate(prediction[0])}
return predictions
predicted_class_name = [labels[k] for k in predicted_class_indices][0] # eh?
def predict(self, wav_filename: str) -> list:
return predicted_class_name, labeled_predictions
def predict(self, wav_filename: str) -> Tuple[str, dict]:
directory, _ = self.create_spectrogram(wav_filename)
result = self._run_predictor(directory)