From b85502cf62888708458621811b2dca61d022bfab Mon Sep 17 00:00:00 2001 From: marcsello Date: Wed, 16 Sep 2020 19:55:21 +0200 Subject: [PATCH] Fixed directory creation --- 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 9e4c194..09a66e4 100644 --- a/cnn_classification_service/cnn_classifier.py +++ b/cnn_classification_service/cnn_classifier.py @@ -43,6 +43,7 @@ class Classifier(object): # 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_basename[:-4]}.jpg") + os.mkdir(os.path.join(target_dir, "unknown")) matplotlib.pyplot.savefig(file_name, dpi=400, bbox_inches='tight', pad_inches=0) matplotlib.pyplot.close() @@ -50,7 +51,7 @@ class Classifier(object): matplotlib.pyplot.close(fig) matplotlib.pyplot.close('all') - return target_dir, file_name + return target_dir, file_name # Az unknown nélkülivel kell visszatérni def _run_predictor(self, directory: str) -> list: predict_generator = self.datagen.flow_from_directory(