fixing tfrecord3

Author: zsdonghao

example/tutorial_tfrecord3.py

@@ -66,7 +66,7 @@ def _bytes_feature_list(values):
 
 processed_capts, img_capts = [], []
 for idx in range(len(caption_data['images'])):
-    img_capt = caption_data['annotations'][idx]['caption']
+    img_capt = caption_data['images'][idx]['caption']
     img_capts.append(img_capt)
     processed_capts.append(tl.nlp.process_sentence(img_capt, start_word="<S>", end_word="</S>"))
 print("Original Captions: %s" % img_capts)
@@ -80,7 +80,7 @@ def _bytes_feature_list(values):
 for idx in range(len(caption_data['images'])):
     # get data
     img_name = caption_data['images'][idx]['file_name']
-    img_capt = '<S> ' + caption_data['annotations'][idx]['caption'] + ' </S>'
+    img_capt = '<S> ' + caption_data['images'][idx]['caption'] + ' </S>'
     img_capt_ids = [vocab.word_to_id(word) for word in img_capt.split(' ')]
     print("%s : %s : %s" % (img_name, img_capt, img_capt_ids))
     img = Image.open(IMG_DIR+img_name)