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

Merge branch 'use_detector_scores' of...

Merge branch 'use_detector_scores' of gitlab-beta.engr.illinois.edu:Vision/GenVQA into use_detector_scores
parents a0564c5f 62963c88
No related branches found
No related tags found
No related merge requests found
...@@ -153,15 +153,9 @@ class eval_mgr(): ...@@ -153,15 +153,9 @@ class eval_mgr():
region_ids, region_ids,
region_paths) region_paths)
if iter%500 == 0: # if iter%500 == 0:
self.write_scores() # self.write_scores()
filename = os.path.join(
self.object_scores_dirname,
'object_predictions.json')
with open(filename,'w') as file:
ujson.dump(self.obj_pred_json,file,indent=4)
def save_image_pred( def save_image_pred(
self, self,
...@@ -231,6 +225,13 @@ class eval_mgr(): ...@@ -231,6 +225,13 @@ class eval_mgr():
labels[:,self.attribute_ids[i]].tolist()) labels[:,self.attribute_ids[i]].tolist())
def write_scores(self): def write_scores(self):
filename = os.path.join(
self.object_scores_dirname,
'object_predictions.json')
with open(filename,'w') as file:
ujson.dump(self.obj_pred_json,file,indent=4)
for i in xrange(10): for i in xrange(10):
filename = os.path.join( filename = os.path.join(
self.attribute_scores_dirname, self.attribute_scores_dirname,
...@@ -384,7 +385,8 @@ def eval( ...@@ -384,7 +385,8 @@ def eval(
# print 'Fall_out: {}'.format(evaluator.get_fall_out()) # print 'Fall_out: {}'.format(evaluator.get_fall_out())
# print 'AP: {}'.format(evaluator.get_ap()) # print 'AP: {}'.format(evaluator.get_ap())
iter+=1 iter+=1
evaluator.write_scores()
# print 'Object accuracy: {}'.format(evaluator.get_object_accuracy()) # print 'Object accuracy: {}'.format(evaluator.get_object_accuracy())
# print 'Recall: {}'.format(evaluator.get_recall()) # print 'Recall: {}'.format(evaluator.get_recall())
# print 'Precision: {}'.format(evaluator.get_precision()) # print 'Precision: {}'.format(evaluator.get_precision())
......
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