Update CALTextModel.py

CALTextModel.py

@@ -606,3 +606,62 @@ def get_sample(ctx0, h_0, k , maxlen, stochastic, training, model):
 
 		return sample, sample_score,sample_att
 
+
+#######################Predict
+    @tf.function(experimental_relax_shapes=True)
+def execute_model(xx,xx_mask,CALTEXT):
+
+  anno = CALTEXT(xx,xx_mask, training=False)
+  hidden_state_0 = CALTEXT.get_hidden_state_0(anno)
+  return anno,hidden_state_0
+
+
+def test_error( images, x_mask):
+  # training=False is only needed if there are layers with different
+  # behavior during training versus inference (e.g. Dropout).
+  batch_loss=0
+  img_ind=1
+  for img_ind in range(len(images)):
+    xx = images[img_ind][tf.newaxis, ... ]
+    xx_mask = x_mask[img_ind][tf.newaxis, ... ]
+    anno,hidden_state_0=execute_model(xx,xx_mask,CALTEXT)
+
+    sample, score,hypalpha=CALTextModel.get_sample(anno, hidden_state_0,10, 130, False, False, CALTEXT)
+
+
+    score = score / np.array([len(s) for s in sample])
+    ss = sample[score.argmin()]
+    img_ind=img_ind+1
+
+    ind=0
+    num=int(len(ss)/2)
+
+    ####   output string
+    ind=0
+    outstr=u''
+    frames = []
+    font = ImageFont.truetype("Jameel Noori Nastaleeq.ttf",60)
+    while (ind<len(ss)-1):
+      k=(len(ss)-2)-ind
+      outstr=outstr+worddicts_r[int(ss[k])]
+      textimg = Image.new('RGB', (1400,100),(255,255,255))
+      drawtext = ImageDraw.Draw(textimg)
+      drawtext.text((20, 20), outstr ,(0,0,0),font=font)
+      fig,axes=plt.subplots(2,1)
+      axes[0].imshow(textimg)
+      axes[0].axis('off')
+      axes[1].axis('off')
+      axes[1].imshow(xx[0,:,:],cmap='gray')
+      visualization=resize(hypalpha[k], (100,800),anti_aliasing=True)
+      axes[1].imshow(255-(255 * visualization), alpha=0.2)
+      plt.axis('off')
+
+      plt.savefig('res.png')
+      frames.append(Image.fromarray(cv2.imread('res.png'), 'RGB'))
+      ind=ind+1
+    frame_one = frames[0]
+    frame_one.save("vis.gif", format="GIF", append_images=frames,save_all=True, duration=300, loop=0)
+    gif_image="vis.gif"
+  return outstr,gif_image
+
+