diff --git a/data/cropped_regions_cached_features.py b/data/cropped_regions_cached_features.py index d71f889d4402825759ca7debf470f4d3b13a615f..ce4dd3f4701a6d388f295822318a91ca9af7c110 100644 --- a/data/cropped_regions_cached_features.py +++ b/data/cropped_regions_cached_features.py @@ -208,9 +208,9 @@ class data(): else: label_id = self.object_labels_dict[object] 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 + # else: + # 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