blinddeconv.processing.utils module

Вспомогательные классы и методы для управления изображениями и метриками.

Автор: Юров П.И.

class blinddeconv.processing.utils.Image(original_path: str, is_color: bool)

Базовые классы: object

Класс для управления путями к изображениям и метриками качества.

В последующем называется «связью».

Атрибуты

original_pathstr

Путь к исходному изображению.

blurred_pathOptional[str]

Путь к размытому изображению (буфер).

blurred_arraynp.ndarray

Несколько вариантов одного размытого изображения.

restored_pathsDict[Tuple[str, str], str]

Список путей к восстановленным изображениям.

is_colorbool

Цветное или черно-белое изображение.

psnrDict[Tuple[str, str], float]

Значения PSNR для восстановленных изображений.

ssimDict[Tuple[str, str], float]

Значения SSIM для восстановленных изображений.

algorithmnp.ndarray

Названия использованных алгоритмов восстановления.

filtersDict[str, str]

Названия использованных фильтров зашумления.

kernel_pathsDict[Tuple[str, str], str]

Путь к полученным psf.

original_kernels_pathDict[str, str]

Путь к изначальным psf.

current_filterOptional[str]

Текущий фильтр.

blurred_psnrDict[str, float]

Значение PSNR для смазанных изображений.

blurred_ssimDict[str, float]

Значение SSIM для смазанных изображений.

preprocessed_blurred_pathDict[str, str]

Изображения после выравнивания гистограмм и/или денойзинга.

Заметки

Структура связи:

Основное изображение original_path - то, от чего начинается связь.

От него идет блок смазанных изображений:

• blurred_path - буфер смазанного изображения

• blurred_array - список всех смазанных изображений (кроме буфера)

• original_kernels_path[путь] - список всех ядер

• blurred_psnr, blurred_ssim[путь] - метрики для смазанных изображений

• current_filter, filters - буфер и список фильтров

От каждого смазанного идет блок восстановленных изображений:

• restored_paths[путь, алгоритм] - пути к восстановленным изображениям

• algorithm - все алгоритмы, которые применялись к связи

• psnr, ssim[путь, алгоритм] - метрики восстановленного изображения

• kernel_paths[путь, алгоритм] - восстановленные psf

Класс необходим для того, чтобы автоматически определять последовательность и порядок картинок, к которым применялись фильтры и методы восстановления

__init__(original_path: str, is_color: bool) → None

Инициализация с путем к исходному изображению.

Параметры

original_pathstr

Путь к исходному изображению.

is_colorbool

Флаг цветного изображения.

add_PSNR(psnr: float, blur_path: str, alg_name: str) → None

Добавляет/переопределяет PSNR конкретного восстановленного изображения.

add_SSIM(ssim: float, blur_path: str, alg_name: str) → None

Добавляет/переопределяет SSIM конкретного воссстановленного изображения.

add_algorithm(algorithm: str) → None

Добавляет алгоритм для восстановления изображения в список.

add_blurred_PSNR(psnr: float, blurred_path: str) → None

Добавляет/переопределяет значение psnr для конкретного смазанного изображения.

add_blurred_SSIM(ssim: float, blurred_path: str) → None

Добавляет/переопределяет значений SSIM для конкретного смазанного изображения.

add_kernel(kernel: str, blur_path: str, alg_path: str) → None

Добавляет/переопределяет путь к ядру конкретного метода для смазанного изображения.

add_original_kernel(kernel: str, blur_path: str) → None

Добавляет/переопределяет путь к ядру конкретного смазанного изображения.

add_preprocessed_blurred_path(blurred_path: str, preprocessed_blurred_path: str) → None

Добавляет изображение для предобработки.

add_restored(restored_paths: str, blur_path: str, alg_name: str) → None

Добавляет/переопределяет путь восстановленного изображения.

add_to_current_filter(filter_str: str) → None

Добавляет фильтр к списку в буфере фильтров.

get_PSNR() → Dict[Tuple[str, str], str]

Возвращает список PSNR восстановленных изображений.

get_SSIM() → Dict[Tuple[str, str], str]

Возвращает список SSIM восстановленных изображений.

get_algorithm() → str

Возвращает список алгоритмов, примененных для восстановления.

get_all_blurred_images() → ndarray | None

Возвращает все смазанные изображения как массивы.

get_all_filters() → ndarray | None

Возращает список всех фильтров смаза.

get_blurred() → str

Возвращает путь к смазанному изображению из буфера.

get_blurred_PSNR() → Dict[str, float]

Возвращает список значений PSNR для смазанных изображений.

get_blurred_SSIM() → Dict[str, float]

Возвращает список значений SSIM смазанных изображений.

get_blurred_array() → ndarray

Возвращает список путей к смазанным изображениям (кроме буфера).

get_blurred_image() → ndarray | None

Возвращает смазанное изображение из буфера как массив.

get_color() → bool

Возвращает флаг цвета.

get_current_filter() → str

Возвращает буфер фильтров.

get_filters() → Dict[str, str]

Возвращает список фильтров.

get_kernels() → Dict[Tuple[str, str], str]

Возвращает список путей к ядрам восстановленных изображений.

get_len_algorithms() → int

Возвращает количество примененных алгоритмов.

get_len_filter() → int

Возращает длину общего списка смазанных изображений (не учитывает буфер).

get_original() → str

Возвращает путь к оригинальному изображению.

get_original_image() → ndarray

Возвращает оригинальное изображение как массив.

get_original_kernels() → Dict[str, str]

Возвращает список путей к ядрам смазанных изображений.

get_preprocessed_blurred_path() → Dict[str, str]

Возвращает путь до изображения для предобработки.

get_restored() → str

Возвращает список восстановленных изображений.

load(index: int) → None

Загружает в буфер из общего списка смазанное изображение.

save_filter() → None

Сохраняет буфер смазанного изображения в общий список.

set_PSNR(psnr: Dict[Tuple[str, str], str]) → None

Полностью переопределяет PSNR восстановленных изображений.

set_SSIM(ssim: Dict[Tuple[str, str], str]) → None

Полностью переопределяет SSIM восстановленных изображений.

set_algorithm(algorithm: ndarray) → None

Полностью переопределяет список алгоритмов восстановления.

set_blurred(blurred_path: str | None) → None

Полностью переопределяет буфер смазанных изображений.

set_blurred_PSNR(psnr: Dict[str, float]) → None

Полностью переопределяет PSNR смазанных изображений.

set_blurred_SSIM(ssim: Dict[str, float]) → None

Полностью переопределяет SSIM смазанных изображений.

set_blurred_array(array: ndarray) → None

Полностью переопределяет пути к смазанным изображениям (кроме буфера).

set_current_filter(filter_str: str) → None

Полностью переопределяет буфер фильров.

set_filters(filters: Dict[str, str]) → None

Полностью переопределяет список фильтров.

set_kernels(kernels: Dict[Tuple[str, str], str]) → None

Полностью переопределяет пути к ядрам восстановленных изображений.

set_original(original_path: str) → None

Полностью переопределяет оригинальное изображение.

set_original_kernels(kernels: Dict[str, str]) → None

Полностью переопределяет пути к ядрам смазанных изображений.

set_preprocessed_blurred_path(preprocessed_blurred_path: Dict[str, str]) → None

Переопределяет изображение для предобработки.

set_restored(restored_paths: Dict[Tuple[str, str], str]) → None

Полностью переопределяет восстановленные изображения.

blinddeconv.processing.utils.calculate_metrics(original_image: ndarray, restored_image: ndarray, data_range: float | None = None) → tuple[float, float]

Расчет метрик восстановления.

blinddeconv.processing.utils.float_img_to_int(image: ndarray) → ndarray

Переводит изображение из диапазона [0.0, 1.0] в диапазон [0, 255].

blinddeconv.processing.utils.generate_unique_file_path(directory: Path, filename: str) → Path

Генерация уникального пути к файлу.

blinddeconv.processing.utils.imread(path: str, color: bool) → ndarray | None

Загружает изображение соответствующего цветового формата.

blinddeconv.processing.utils.prepare_image_for_metric(image: ndarray) → ndarray

Подготовка изображения для расчета метрик с нормализацией.