Skip to content
Snippets Groups Projects
Commit 2e980c63 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 e2df6fc2 e0fcf8c9
No related branches found
No related tags found
No related merge requests found
import os
import ujson
import pprint
import math
from collections import namedtuple
from operator import attrgetter
import numpy as np
......@@ -12,9 +13,11 @@ import constants
pp = pprint.PrettyPrinter(indent=4)
if __name__=='__main__':
print 'Reading {}'.format(constants.regions_json)
with open(constants.regions_json) as file:
all_regions_data = ujson.load(file)
print 'Reading {}'.format(constants.genome_train_region_ids)
with open(constants.genome_train_region_ids) as file:
train_region_ids = ujson.load(file)
......@@ -22,6 +25,7 @@ if __name__=='__main__':
constants.region_object_scores_dirname,
'object_predictions.json')
print 'Reading {}'.format(obj_pred_json)
with open(obj_pred_json) as file:
obj_pred_data = ujson.load(file)
......@@ -78,7 +82,18 @@ if __name__=='__main__':
for record in sorted_records:
freq.append(record.train_freq)
acc.append(record.acc)
plt.plot(freq,acc)
acc_by_freq = {
'freq': freq,
'acc': acc,
}
acc_by_freq_filename = os.path.join(
constants.region_object_scores_dirname,
'acc_by_freq.json')
with open(acc_by_freq_filename,'w') as file:
ujson.dump(acc_by_freq,file)
plt.plot(freq[200:800],acc[200:800])
plot_filename = os.path.join(
constants.region_object_scores_dirname,
'obj_acc_vs_freq.pdf')
......
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