Skip to content

GitLab

Commit fb10543d authored by Vishnu Banna's avatar Vishnu Banna
Browse files

kmeans

parent dff8897d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 5 deletions
official/vision/beta/projects/yolo/configs/yolo.py
View file @ fb10543d
...@@ -189,7 +189,6 @@ class AnchorBoxes(hyperparams.Config): ...@@ -189,7 +189,6 @@ class AnchorBoxes(hyperparams.Config):
generate_anchors: bool = False generate_anchors: bool = False
scaling_mode: str = "sqrt_log" scaling_mode: str = "sqrt_log"
box_generation_mode: str = "per_level" box_generation_mode: str = "per_level"
use_validation_data: bool = True
def get(self, min_level, max_level): def get(self, min_level, max_level):
"""Distribute them in order to each level. """Distribute them in order to each level.
......
official/vision/beta/projects/yolo/tasks/yolo.py
View file @ fb10543d
...@@ -74,10 +74,7 @@ class YoloTask(base_task.Task): ...@@ -74,10 +74,7 @@ class YoloTask(base_task.Task):
anchor_cfg = self.task_config.model.anchor_boxes anchor_cfg = self.task_config.model.anchor_boxes
backbone = self.task_config.model.backbone.get() backbone = self.task_config.model.backbone.get()
if anchor_cfg.use_validation_data: dataset = self.task_config.train_data
dataset = self.task_config.validation_data
else:
dataset = self.task_config.train_data
decoder = self._get_data_decoder(dataset) decoder = self._get_data_decoder(dataset)
num_anchors = backbone.max_level - backbone.min_level + 1 num_anchors = backbone.max_level - backbone.min_level + 1
......
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