From a0c4b8aa58ed6e9123fc2c8bb974d0ad369ca84a Mon Sep 17 00:00:00 2001 From: tgupta6 <tgupta6@illinois.edu> Date: Sat, 8 Oct 2016 15:11:23 -0500 Subject: [PATCH] no fractional object prob in gt label vector --- data/cropped_regions_cached_features.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/data/cropped_regions_cached_features.py b/data/cropped_regions_cached_features.py index ce4dd3f..2aa8617 100644 --- a/data/cropped_regions_cached_features.py +++ b/data/cropped_regions_cached_features.py @@ -212,7 +212,8 @@ class data(): # label_id = self.object_labels_dict[_unknown_token] # object_label_encoding[0,label_id] = 1.0 - return object_label_encoding/np.sum(object_label_encoding), object_labels + #return object_label_encoding/np.sum(object_label_encoding), object_labels + return object_label_encoding, object_labels def get_attribute_label(self, sample): # Attribute is turned on if it is present -- GitLab