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

Fixed float conversion
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Pünkösd Marcell
2021-06-14 03:28:32 +02:00
parent f752668cb5
commit c757b838dd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cnn_classification_service/cnn_classifier.py
View File

@@ -77,7 +77,7 @@ class Classifier(object):
} }
labels = dict((v, k) for k, v in labels.items()) labels = dict((v, k) for k, v in labels.items())
labeled_predictions = {labels[i]: p for i, p in enumerate(prediction[0])} labeled_predictions = {labels[i]: float(p) for i, p in enumerate(prediction[0])}
predicted_class_name = [labels[k] for k in predicted_class_indices][0] # eh? predicted_class_name = [labels[k] for k in predicted_class_indices][0] # eh?