Commit 7dcb0690 authored by Rudolf Chrispens's avatar Rudolf Chrispens
Browse files

finished first part of python notebook jasons part to go

parent 8ffd4851

caption_lib/LSTM.ipynb

+27 −5

File changed.

Preview size limit exceeded, changes collapsed.

caption_lib/lstm/Validators/coco_validator.py

+22 −2
Original line number Diff line number Diff line
@@ -91,9 +91,18 @@ class coco_validator(object): 
                for n in range(_config.validator.attention_sample_number):

                    print("#>\tSampled Caption: %s" % ' '.join(decoded[n]))







                    # Plot original image

                    print("#>\tPlot attention image:" + image_path + "/" + str(image_files[n]))

                    img = ndimage.imread(image_path + "/" + str(image_files[n]))



                    #image big:

                    plt.imshow(img)

                    plt.axis('off')

                    plt.show()

                    plt.close()



                    plt.subplot(4, 5, 1)



                    plt.imshow(img)
@@ -122,7 +131,15 @@ class coco_validator(object): 
                        plt.imshow(alp_img, alpha=0.5)

                        plt.axis('off')



                        if words[t] == '<END>':

                            break





                    plt.figure(figsize=(12, 12))

                    plt.show()

                    plt.close()









            if save_sampled_captions:
@@ -148,6 +165,9 @@ class coco_validator(object): 
                captionlist_for_eval = []



                for numb, caption in enumerate(all_decoded[:num_iter * self.batch_size]):

                    captionlist_for_eval.append({'caption': ' '.join(caption), 'image_id': int(image_idxs[numb])})

                    if i % 5 == 0:

                        captionlist_for_eval.append({'caption': ' '.join(caption), 'image_id': int(image_idxs[numb]), 'image_name': data['file_names'][numb] })



                self.save_json(captionlist_for_eval, path + split + ".candidate.captions.json")



                self.save_json(captionlist_for_eval[:num_iter * self.batch_size], path + split + ".candidate.captions.json")