Exemple #1
0
  def __init__(self, params, logging_dir: Optional[str] = None):
    super().__init__(params, logging_dir)
    self.coco_metric = None
    self._loss_fn = None
    self._model = None
    self._coco_91_to_80 = False
    self._metrics = []

    # globally set the random seed
    preprocessing_ops.set_random_seeds(seed=params.seed)
    return
Exemple #2
0
    def __init__(self, params, logging_dir: Optional[str] = None):
        super().__init__(params, logging_dir)
        self.coco_metric = None
        self._loss_fn = None
        self._model = None
        self._coco_91_to_80 = False
        self._metrics = []

        # globally set the random seed
        preprocessing_ops.set_random_seeds(seed=params.seed)

        if self.task_config.model.anchor_boxes.generate_anchors:
            self.generate_anchors()
        return