FBAGSTM's picture
STM32 AI Experimentation Hub
747451d
# /*---------------------------------------------------------------------------------------------
# * 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 .anchor_boxes_utils import get_sizes_ratios_ssd_v1, get_sizes_ratios_ssd_v2, get_fmap_sizes, gen_anchors, get_anchor_boxes
from .parse_config import parse_dataset_section, parse_preprocessing_section, parse_data_augmentation_section, get_config
from .models_mgt import ai_runner_invoke, change_model_input_shape, change_yolo_model_number_of_classes, \
change_yolo_x_model_number_of_classes
from .gen_h_file import gen_h_user_file_h7, gen_h_user_file_n6, gen_h_user_file_n6_onnx_ssd, gen_h_user_file_n6_onnx_yolod
from .objdet_metrics import ObjectDetectionMetricsData, calculate_average_metrics, calculate_objdet_metrics
from .bounding_boxes_utils import bbox_center_to_corners_coords, bbox_corners_to_center_coords, bbox_normalized_to_abs_coords, \
bbox_abs_to_normalized_coords, calculate_box_wise_iou, plot_bounding_boxes