File size: 2,298 Bytes
747451d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | # /*---------------------------------------------------------------------------------------------
# * 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 .logs_utils import mlflow_ini, log_to_file, log_last_epoch_history, LRTensorBoard
from .cfg_utils import (check_attributes, collect_callback_args, check_config_attributes, get_random_seed, replace_none_string,
expand_env_vars, postprocess_config_dict, parse_tools_section, parse_benchmarking_section, parse_model_section,
parse_quantization_section, parse_evaluation_section, parse_top_level, parse_general_section,
parse_random_periodic_resizing, parse_training_section, parse_prediction_section, parse_compression_section,
parse_deployment_section, parse_mlflow_section, check_hardware_type, get_class_names_from_file,
aspect_ratio_dict, color_mode_n6_dict, check_model_file_extension, download_file)
from .gpu_utils import set_gpu_memory_limit, inc_gpu_mode, check_training_determinism, get_mem_consumption, \
gpu_benchmark
from .models_utils import ai_interp_input_quant, ai_interp_outputs_dequant, ai_runner_interp, get_model_name, \
get_model_name_and_its_input_shape, check_model_support, check_attribute_value, \
transfer_pretrained_weights, model_summary, count_h5_parameters, count_tflite_parameters, \
tf_dataset_to_np_array, compute_confusion_matrix, torch_dataset_to_np_array
from .visualize_utils import vis_training_curves, plot_confusion_matrix, display_figures, compute_confusion_matrix2, \
compute_multilabel_confusion_matrices, plot_multilabel_confusion_matrices
from .pt_logger import LOGGER
from .cfg_utils import flatten_config
|