remove object and group from object_labels.json (modification in visual_genome_parser.py)

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()