diff --git a/visual_genome_parser.py b/visual_genome_parser.py index ed6bcdeef65933e9590a3fe03d55c28b0ef66a17..ff447c8917862e54c658694543d2b2f9f7ddeabe 100644 --- a/visual_genome_parser.py +++ b/visual_genome_parser.py @@ -255,7 +255,16 @@ def top_k_object_labels(k): _raw_object_labels) with open(raw_object_labels_filename, 'r') as file: raw_object_labels = json.load(file) - + + print "Number of times 'object' appears: {}".format( + raw_object_labels.pop('object',None)) + print "Number of times 'objects' appears: {}".format( + raw_object_labels.pop('objects',None)) + print "Number of times 'group' appears: {}".format( + raw_object_labels.pop('group',None)) + print "Number of times 'groups' appears: {}".format( + raw_object_labels.pop('groups',None)) + sorted_raw_object_labels = \ [key for key, value in sorted(raw_object_labels.items(), key = operator.itemgetter(1), @@ -454,7 +463,7 @@ def select_vocab_subset(k): vocab_filename = os.path.join(_outdir, _vocab) with open(vocab_filename, 'r') as file: vocab = json.load(file) - + sorted_vocab = \ [key for key, value in sorted(vocab.items(), key = operator.itemgetter(1), @@ -701,10 +710,10 @@ if __name__=='__main__': # _vqa_train_anno, # _vqa_val_anno, # _genome_train_subset_region_ids) - add_hypernyms( - _regions_with_labels, - _regions_with_hypernym_labels, - _object_labels_to_hypernyms) + # add_hypernyms( + # _regions_with_labels, + # _regions_with_hypernym_labels, + # _object_labels_to_hypernyms) # generate_md5hash()