From 726009a257bcfd443f5534cdc51db520c1a48c53 Mon Sep 17 00:00:00 2001
From: tgupta6 <tgupta6@illinois.edu>
Date: Sun, 9 Oct 2016 15:32:19 -0500
Subject: [PATCH] remove object and group from object_labels.json (modification
in visual_genome_parser.py)
---
visual_genome_parser.py | 21 +++++++++++++++------
1 file changed, 15 insertions(+), 6 deletions(-)
diff --git a/visual_genome_parser.py b/visual_genome_parser.py
index ed6bcde..ff447c8 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()
--
GitLab