def show_errors(error_infos, fonts): length = len(error_infos) labels = len(fonts) for i in xrange(length): text, pred = error_infos[i] index = pred.index(max(pred)) for j in xrange(labels): axis = plt.subplot(length, labels, i * labels + j + 1) axis.axis('off') font = fonts[j] image, _ = draw_font(text, font, mode='test') if index == j: plt.title(str(pred)) plt.imshow(image) plt.show()
def generator_images(texts, fonts): for text in texts: for font in fonts: image, _ = draw_font(text, font, mode='test') yield image, text