Skip to content
Snippets Groups Projects
Commit 98f2b8b5 authored by tgupta6's avatar tgupta6
Browse files

if unknown then no gradient (changed object labels in cropped_regions_cached_features.py)

parent f88c7cd9
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
data/cropped_regions_cached_features.py
+ 3
− 3
View file @ 98f2b8b5
...@@ -208,9 +208,9 @@ class data(): ...@@ -208,9 +208,9 @@ class data():
else: else:
label_id = self.object_labels_dict[object] label_id = self.object_labels_dict[object]
object_label_encoding[0,label_id] = 1.0 object_label_encoding[0,label_id] = 1.0
else: # else:
label_id = self.object_labels_dict[_unknown_token] # label_id = self.object_labels_dict[_unknown_token]
object_label_encoding[0,label_id] = 1.0 # 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
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment