Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "c1a81c6fe3e56e7e185a890cb30a5e57462e9a3f"
Commit f35907fd authored by Vishnu Banna's avatar Vishnu Banna
Browse files

head update

parent e3f09134
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/projects/yolo/modeling/heads/yolo_head.py
View file @ f35907fd
...@@ -125,7 +125,7 @@ class YoloHead(tf.keras.layers.Layer): ...@@ -125,7 +125,7 @@ class YoloHead(tf.keras.layers.Layer):
def num_boxes(self): def num_boxes(self):
if self._min_level is None or self._max_level is None: if self._min_level is None or self._max_level is None:
raise Exception( raise Exception(
'model has to be built before number of boxes can be determined') 'Model has to be built before number of boxes can be determined.')
return (self._max_level - self._min_level + 1) * self._boxes_per_level return (self._max_level - self._min_level + 1) * self._boxes_per_level
@property @property
......
Markdown is supported
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