diff --git a/object_attribute_classifier_cached_features/compute_AP.py b/object_attribute_classifier_cached_features/compute_AP.py index c2dc5dbc80879e423eac3b40b3b624da9f11192a..094304746bde515235d8b94dcc5c109b065bf0a5 100644 --- a/object_attribute_classifier_cached_features/compute_AP.py +++ b/object_attribute_classifier_cached_features/compute_AP.py @@ -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')