From 1b955c608aa5c93ce25ee861bb60dea066f45f55 Mon Sep 17 00:00:00 2001 From: kaihuatang Date: Fri, 14 Aug 2020 16:26:22 +0800 Subject: [PATCH] no message --- maskrcnn_benchmark/data/datasets/visual_genome.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/maskrcnn_benchmark/data/datasets/visual_genome.py b/maskrcnn_benchmark/data/datasets/visual_genome.py index 9884cbf..25ef411 100644 --- 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,