Spaces:
Sleeping
Sleeping
Update visualization.py
Browse files- visualization.py +1 -1
visualization.py
CHANGED
@@ -70,6 +70,6 @@ def print_top_classes(original_image, **kwargs):
|
|
70 |
for cls_idx in class_indices:
|
71 |
output_string = '{} : {}'.format(cls_idx, CLS2IDX[cls_idx])
|
72 |
# output_string += ' ' * (max_str_len - len(CLS2IDX[cls_idx])) + '\t\t'
|
73 |
-
output_string += 'value = {:.3f} prob = {:.1f}%'.format(predictions[0, cls_idx], 100 * prob[0, cls_idx])
|
74 |
output.append(output_string)
|
75 |
return output
|
|
|
70 |
for cls_idx in class_indices:
|
71 |
output_string = '{} : {}'.format(cls_idx, CLS2IDX[cls_idx])
|
72 |
# output_string += ' ' * (max_str_len - len(CLS2IDX[cls_idx])) + '\t\t'
|
73 |
+
output_string += ' value = {:.3f} prob = {:.1f}%'.format(predictions[0, cls_idx], 100 * prob[0, cls_idx])
|
74 |
output.append(output_string)
|
75 |
return output
|