# /*---------------------------------------------------------------------------------------------
#  * 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