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
Hide whitespace changes
Inline Side-by-side
object_attribute_classifier_cached_features/compute_AP.py
+ 6
− 4
View file @ b918ff2a
...@@ -3,7 +3,7 @@ import pdb ...@@ -3,7 +3,7 @@ import pdb
import ujson import ujson
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np import numpy as np
import constants
def compute_AP(precision, recall): def compute_AP(precision, recall):
ap = 0.0 ap = 0.0
...@@ -18,9 +18,11 @@ if __name__=='__main__': ...@@ -18,9 +18,11 @@ if __name__=='__main__':
AP = [] AP = []
meanAP = 0.0 meanAP = 0.0
for i in xrange(10): for i in xrange(10):
dirname = '/home/tanmay/Code/GenVQA/Exp_Results/VisualGenome/' + \ # dirname = '/home/tanmay/Code/GenVQA/Exp_Results/VisualGenome/' + \
'object_attribute_classifier_large_images/' + \ # 'object_attribute_classifier_large_images/' + \
'object_attribute_classifiers/attribute_scores/' # 'object_attribute_classifiers/attribute_scores/'
dirname = constants.region_attribute_scores_dirname
labels_filename = os.path.join(dirname, 'labels_' + str(i) + '.json') labels_filename = os.path.join(dirname, 'labels_' + str(i) + '.json')
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment