From 5a257b12cd5e6e56809ee1593abd59bbed0dc1d2 Mon Sep 17 00:00:00 2001 From: tgupta6 <tgupta6@illinois.edu> Date: Sun, 9 Oct 2016 21:15:27 -0500 Subject: [PATCH] multi hot object label vector in cropped_regions_cached_features.py --- data/cropped_regions_cached_features.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/data/cropped_regions_cached_features.py b/data/cropped_regions_cached_features.py index 2aa8617..ca3b99f 100644 --- a/data/cropped_regions_cached_features.py +++ b/data/cropped_regions_cached_features.py @@ -204,10 +204,11 @@ class data(): if object_labels: for object in object_labels: if object not in self.object_labels_dict: - label_id = self.object_labels_dict[_unknown_token] + pass + #label_id = self.object_labels_dict[_unknown_token] else: label_id = self.object_labels_dict[object] - object_label_encoding[0,label_id] = 1.0 + object_label_encoding[0,label_id] = 1.0 # else: # label_id = self.object_labels_dict[_unknown_token] # object_label_encoding[0,label_id] = 1.0 -- GitLab