diff --git a/data/cropped_regions_cached_features.py b/data/cropped_regions_cached_features.py index 2aa8617a4518797e44ca984303814c47feb274c4..ca3b99f16a485eeb031d53564e4ecc12a6611c94 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