Source Code Documentation

• VISWIR_vQuasar module
  • Workflow
  • main()
• common package
  • Submodules
  • common.config_loader module
    • config_loader.py
    • ConfigError
    • load_all_configs()
    • load_config()
    • load_optuna_search_space()
    • resolve_config_path()
    • validate_config()
  • common.datatypes module
    • FusionConfig
      • FusionConfig.visible_folder
      • FusionConfig.swir_folder
      • FusionConfig.output_folder
      • FusionConfig.ref_image_path
      • FusionConfig.ground_truth_path
      • FusionConfig.mode
      • FusionConfig.run_detection
      • FusionConfig.ground_truth_path
      • FusionConfig.mode
      • FusionConfig.output_folder
      • FusionConfig.ref_image_path
      • FusionConfig.run_detection
      • FusionConfig.swir_folder
      • FusionConfig.visible_folder
    • FusionTask
      • FusionTask.visible_path
      • FusionTask.swir_path
      • FusionTask.ref_image_path
      • FusionTask.ground_truth_path
      • FusionTask.params
      • FusionTask.save_output
      • FusionTask.run_detection
      • FusionTask.output_dir
      • FusionTask.ground_truth_path
      • FusionTask.output_dir
      • FusionTask.params
      • FusionTask.ref_image_path
      • FusionTask.run_detection
      • FusionTask.save_output
      • FusionTask.swir_path
      • FusionTask.visible_path
    • OptunaConfig
      • OptunaConfig.n_trials
      • OptunaConfig.n_jobs
      • OptunaConfig.sampler
      • OptunaConfig.pruner
      • OptunaConfig.timeout
      • OptunaConfig.n_jobs
      • OptunaConfig.n_trials
      • OptunaConfig.pruner
      • OptunaConfig.sampler
      • OptunaConfig.timeout
    • OptunaResult
      • OptunaResult.visible_path
      • OptunaResult.swir_path
      • OptunaResult.params
      • OptunaResult.metrics_fusion
      • OptunaResult.error
      • OptunaResult.error
      • OptunaResult.metrics_fusion
      • OptunaResult.params
      • OptunaResult.swir_path
      • OptunaResult.visible_path
    • ProcessResult
      • ProcessResult.visible_path
      • ProcessResult.swir_path
      • ProcessResult.ground_truth_path
      • ProcessResult.params
      • ProcessResult.metrics_fusion
      • ProcessResult.metrics_visible
      • ProcessResult.metrics_swir
      • ProcessResult.error
      • ProcessResult.error
      • ProcessResult.ground_truth_path
      • ProcessResult.metrics_fusion
      • ProcessResult.metrics_swir
      • ProcessResult.metrics_visible
      • ProcessResult.params
      • ProcessResult.swir_path
      • ProcessResult.visible_path
  • common.logger module
    • Features
    • Usage
  • common.results_db module
    • FusionResult
      • FusionResult.id
      • FusionResult.visible_img
      • FusionResult.swir_img
      • FusionResult.ref_img
      • FusionResult.grd_tr
      • FusionResult.alpha
      • FusionResult.beta
      • FusionResult.level
      • FusionResult.gamma
      • FusionResult.metrics_f
      • FusionResult.metrics_v
      • FusionResult.metrics_s
      • FusionResult.error
      • FusionResult.alpha
      • FusionResult.beta
      • FusionResult.error
      • FusionResult.gamma
      • FusionResult.grd_tr
      • FusionResult.id
      • FusionResult.level
      • FusionResult.metrics_f
      • FusionResult.metrics_s
      • FusionResult.metrics_v
      • FusionResult.ref_img
      • FusionResult.swir_img
      • FusionResult.visible_img
    • get_session()
    • save_result_to_db()
  • common.ui module
    • print_license_info()
    • print_viswir_header()
  • Module contents
    • common package
• fusion package
  • Subpackages
    • fusion.NIQE package
      • Submodules
      • fusion.NIQE.niqe module
      • Module contents
  • Submodules
  • fusion.detection_module module
    • iou()
    • parse_voc_annotations()
    • prepare_image_for_yolo()
    • run_yolo_and_compute_f1()
    • save_annotated_float64_image_as_uint16()
    • save_predictions_as_voc_xml()
  • fusion.functions module
    • calculate_levels()
    • downsample()
    • gaussian_pyramid()
    • laplacian_pyramid()
    • local_std()
    • local_visibility()
    • my_upsample()
    • pyramid_filter()
    • reconstruct_laplacian_pyramid()
    • reconstruct_laplacian_pyramid_2()
    • reconstruct_without_laplacian()
  • fusion.fusion module
    • process_image()
    • viswir_core_fusion()
  • fusion.metrics module
    • calculate_brisque()
    • calculate_correlation()
    • calculate_entropy()
    • calculate_entropy_normalized()
    • calculate_ergas()
    • calculate_gmsd()
    • calculate_gmsd_color()
    • calculate_mad()
    • calculate_mad_color()
    • calculate_mean_gradient()
    • calculate_mean_intensity()
    • calculate_ms_ssim_pytorch()
    • calculate_ms_ssim_pytorch_color()
    • calculate_mse()
    • calculate_niqe_metric()
    • calculate_piqe()
    • calculate_psnr()
    • calculate_rmse()
    • calculate_sam()
    • calculate_snr()
    • calculate_snr_color()
    • calculate_snr_color_per_channel()
    • calculate_spatial_frequency_color()
    • calculate_spatial_frequency_grey()
    • calculate_ssim()
    • calculate_std()
    • calculate_uqi()
    • calculate_vif()
    • compute_all_metrics()
    • register_metric()
    • safe()
  • fusion.utils module
    • configure_logger()
    • display_image()
    • ensure_grayscale()
    • ensure_range_255()
    • from_grey_to_rgb_array()
    • generate_parameters_json()
    • generate_parameters_json_return()
    • get_image_shape()
    • load_image_opencv()
    • load_image_ref()
    • load_image_ref_skimage()
    • load_image_skimage_core()
    • load_parameters_json()
    • normalize_image()
    • prepare_ground_truth_list()
    • prepare_ground_truth_list_vLegacy()
    • preprocess_images_for_metrics()
    • print_image_info()
    • safe_float()
    • save_float64_image_as_uint16()
    • save_image()
    • str_to_bool_strict()
    • to_float32()
    • to_grayscale_array_OLD()
    • to_grayscale_array_cv2()
    • to_grayscale_array_manual()
    • to_grayscale_array_skimage()
    • to_rgb_array()
    • visualize_pyramid()
    • wait_if_paused()
  • Module contents
    • fusion package
• optimization package
  • Submodules
  • optimization.objective module
    • objective()
    • objective_detection()
    • suggest_params()
  • optimization.optuna_runner module
    • clear_cache_callback()
    • normalize()
    • optimize_parameters_for_group_v2()
    • process_folder_optuna()
    • tchebycheff_distance()
  • optimization.optuna_wrapper module
    • clear_cache_callback()
    • normalize()
    • optimize_parameters_for_group_v2()
    • process_folder_optuna()
    • tchebycheff_distance()
  • optimization.samplers module
    • get_pruner()
    • get_sampler()
  • optimization.visualization module
    • visualize_study()
  • Module contents
    • Optimization package
      • Submodules
      • Features
• processing package
  • Submodules
  • processing.batch_runner module
    • process_folder()
    • process_image_wrapper()
  • processing.interruption module
    • save_last_params()
    • signal_handler()
  • processing.sql_runner module
    • process_folder_sql()
  • processing.task_manager module
    • batchify_tasks()
    • generate_tasks_in_memory()
  • Module contents
    • processing package
• realtime package
  • Submodules
  • realtime.fast_config module
    • load_fast_config()
  • realtime.fast_detection module
    • FastDetector
      • FastDetector.draw()
      • FastDetector.draw_label()
      • FastDetector.predict()
  • realtime.fast_fusion_runner module
    • FastFusionPipeline
      • FastFusionPipeline.cfg
      • FastFusionPipeline.detector
      • FastFusionPipeline.run_detection
      • FastFusionPipeline.defaults
      • FastFusionPipeline.run()
    • read_image_swir()
    • read_image_visible()
  • Module contents
    • realtime package