Skip to content
Snippets Groups Projects
Commit 5a257b12 authored by tgupta6's avatar tgupta6
Browse files

multi hot object label vector in cropped_regions_cached_features.py

parent 2216f3bc
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment