no message

moonshot-s · Aug 14, 2020 · 1b955c6 · 1b955c6
1 parent 2e59892
commit 1b955c6
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/maskrcnn_benchmark/data/datasets/visual_genome.py b/maskrcnn_benchmark/data/datasets/visual_genome.py
@@ -49,14 +49,13 @@ def __init__(self, split, img_dir, roidb_file, dict_file, image_file, transforms
         self.filter_duplicate_rels = filter_duplicate_rels and self.split == 'train'
         self.transforms = transforms
 
+        self.ind_to_classes, self.ind_to_predicates, self.ind_to_attributes = load_info(dict_file) # contiguous 151, 51 containing __background__
+        self.categories = {i : self.ind_to_classes[i] for i in range(len(self.ind_to_classes))}
+
         self.custom_eval = custom_eval
         if self.custom_eval:
             self.get_custom_imgs(custom_path)
         else:
-            self.ind_to_classes, self.ind_to_predicates, self.ind_to_attributes = load_info(dict_file) # contiguous 151, 51 containing __background__
-
-            self.categories = {i : self.ind_to_classes[i] for i in range(len(self.ind_to_classes))}
-
             self.split_mask, self.gt_boxes, self.gt_classes, self.gt_attributes, self.relationships = load_graphs(
                 self.roidb_file, self.split, num_im, num_val_im=num_val_im,
                 filter_empty_rels=filter_empty_rels,