File size: 900 Bytes
747451d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | # /*---------------------------------------------------------------------------------------------
# * Copyright (c) 2022 STMicroelectronics.
# * All rights reserved.
# * This software is licensed under terms that can be found in the LICENSE file in
# * the root directory of this software component.
# * If no LICENSE file comes with this software, it is provided AS-IS.
# *--------------------------------------------------------------------------------------------*/
from .utils.callbacks import get_callbacks
from .utils.ssd.ssd_anchor_matching import match_gt_anchors
from .utils.ssd.ssd_loss import ssd_focal_loss
from .utils.ssd.ssd_train_model import SSDTrainingModel
from .utils.yolo.yolo_loss import get_detector_mask, yolo_loss
from .utils.yolo.yolo_train_model import YoloTrainingModel
from .utils.yolo.yolo_x_train_model import YoloXTrainingModel
from .od_trainer import ODTrainer
|