Skip to content
Snippets Groups Projects
Commit b918ff2a authored by tgupta6's avatar tgupta6
Browse files

compute_AP uses constants file

parent 8980e4f7
No related branches found
No related tags found
No related merge requests found
......@@ -3,7 +3,7 @@ import pdb
import ujson
import matplotlib.pyplot as plt
import numpy as np
import constants
def compute_AP(precision, recall):
ap = 0.0
......@@ -18,9 +18,11 @@ if __name__=='__main__':
AP = []
meanAP = 0.0
for i in xrange(10):
dirname = '/home/tanmay/Code/GenVQA/Exp_Results/VisualGenome/' + \
'object_attribute_classifier_large_images/' + \
'object_attribute_classifiers/attribute_scores/'
# dirname = '/home/tanmay/Code/GenVQA/Exp_Results/VisualGenome/' + \
# 'object_attribute_classifier_large_images/' + \
# 'object_attribute_classifiers/attribute_scores/'
dirname = constants.region_attribute_scores_dirname
labels_filename = os.path.join(dirname, 'labels_' + str(i) + '.json')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment