WwYc commited on
Commit
91913b5
·
verified ·
1 Parent(s): 071c27b

Update visualization.py

Browse files
Files changed (1) hide show
  1. 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