From 65adfa97963283814740351d63dd1a3832fa572c Mon Sep 17 00:00:00 2001 From: Zhe Wang Date: Fri, 1 Dec 2017 16:14:33 -0800 Subject: [PATCH] Add files via upload --- generate_attention.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 generate_attention.py diff --git a/generate_attention.py b/generate_attention.py new file mode 100644 index 0000000..9c9258f --- /dev/null +++ b/generate_attention.py @@ -0,0 +1,33 @@ +#-*- coding: utf-8 -*- +# 64.72% # +import tensorflow as tf +import pandas as pd +import numpy as np +import os, h5py, sys, argparse +import ipdb +import time +import math +import cv2 +import codecs, json +# from tensorflow.python.ops import rnn_cell +from sklearn.metrics import average_precision_score +import pdb +import spacy +from random import seed +import itertools +from numpy import linalg as LA + +tf.reset_default_graph() +random_seed = 320 +np.random.seed(random_seed) +tf.set_random_seed(random_seed) + +triple = json.load(open('/home/liangjic/BMVC/VQA_20170301/vqa_raw_train.json')) +print triple[210716] +for xid in xrange(len(triple)): + if xid % 4 == 0 and triple[xid]['ans'] == 0: + print xid + break + if xid % 4 > 0 and triple[xid]['ans'] == 1: + print xid + break