From b918ff2afa2df134ab872a5f026912b232c670d2 Mon Sep 17 00:00:00 2001
From: tgupta6 <tgupta6@illinois.edu>
Date: Mon, 29 Aug 2016 14:45:22 -0500
Subject: [PATCH] compute_AP uses constants file
---
.../compute_AP.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/object_attribute_classifier_cached_features/compute_AP.py b/object_attribute_classifier_cached_features/compute_AP.py
index c2dc5db..0943047 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')
--
GitLab