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

changed k to 100 in obj atr eval

parent 2d88e085
No related branches found
No related tags found
No related merge requests found
...@@ -126,10 +126,12 @@ class eval_mgr(): ...@@ -126,10 +126,12 @@ class eval_mgr():
# self.eval_object_accuracy( # self.eval_object_accuracy(
# eval_vars_dict['object_prob'], # eval_vars_dict['object_prob'],
# labels['objects']) # labels['objects'])
k = 100
self.top_k_accuracy( self.top_k_accuracy(
eval_vars_dict['object_prob'], eval_vars_dict['object_prob'],
labels['objects'], labels['objects'],
20) k)
self.eval_attribute_pr( self.eval_attribute_pr(
eval_vars_dict['attribute_prob'], eval_vars_dict['attribute_prob'],
...@@ -139,9 +141,9 @@ class eval_mgr(): ...@@ -139,9 +141,9 @@ class eval_mgr():
eval_vars_dict['attribute_prob'], eval_vars_dict['attribute_prob'],
labels['attributes']) labels['attributes'])
pred_obj_labels = self.get_top_k_labels(eval_vars_dict['object_prob'],20) pred_obj_labels = self.get_top_k_labels(eval_vars_dict['object_prob'],k)
gt_obj_labels = self.get_gt_labels(labels['objects']) gt_obj_labels = self.get_gt_labels(labels['objects'])
pred_atr_labels = self.get_top_k_labels(eval_vars_dict['attribute_prob'],20,'atr') pred_atr_labels = self.get_top_k_labels(eval_vars_dict['attribute_prob'],k,'atr')
gt_atr_labels = self.get_gt_labels(labels['attributes'],'atr') gt_atr_labels = self.get_gt_labels(labels['attributes'],'atr')
region_paths = self.get_region_paths(image_ids, region_ids) region_paths = self.get_region_paths(image_ids, region_ids)
......
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