blinddeconv.scripts package

Submodules

Module contents

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

Модули:: kernel_generator: Генерация ядер размытия (PSF) dataset_generator: Генерация датасета с искажениями

class blinddeconv.scripts.DatasetGenerator(processing_instance: Processing, input_dir: str = 'images_dataset/original', output_dir: str = 'images_dataset/distorted', kernel_dir: str = 'images_dataset/ground_truth_filters', kernel_data_dir: str = 'images_dataset/kernel_data')[исходный код]

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

Генератор датасета для экспериментов по слепой деконволюции.

Класс автоматически создаёт искажённые изображения с различными комбинациями размытия и шума, сохраняет ядра (PSF) и связывает все данные через Processing.bind().

Параметры

processing_instancepr.Processing: Экземпляр класса Processing для связывания данных.
input_dirstr, optional: Путь к папке с исходными изображениями.
output_dirstr, optional: Путь к папке для сохранения искажённых изображений.
kernel_dirstr, optional: Путь к папке для сохранения PNG-изображений ядер.
kernel_data_dirstr, optional: Путь к папке для сохранения NPY-файлов ядер.

Атрибуты

blur_configsList[Dict]: Конфигурации фильтров размытия.
noise_configsList[Dict]: Конфигурации шумовых фильтров.

generate_and_save_kernels() → Tuple[Dict[str, Dict], List][исходный код]

Генерация и сохранение всех ядер размытия.

Возвращает

kernel_pathsDict[str, Dict]: Словарь с путями к файлам ядер.
visualizationsList[Tuple[str, np.ndarray]]: Список данных для визуализации.

process_and_bind_images() → None[исходный код]

Основной метод: генерация датасета.

Создаёт искажённые изображения со всеми комбинациями размытия и шума, сохраняет результаты и связывает через Processing.

class blinddeconv.scripts.KernelGenerator(kernel_dir: str = 'images_dataset/ground_truth_filters', kernel_data_dir: str = 'images_dataset/kernel_data', kernel_size: int = 51)[исходный код]

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

Генератор ядер размытия (PSF) для экспериментов по деконволюции.

Параметры

kernel_dirstr: Путь для сохранения PNG-изображений ядер.
kernel_data_dirstr: Путь для сохранения NPY-файлов ядер.
kernel_sizeint: Размер генерируемых ядер.

generate_all() → Dict[str, ndarray][исходный код]

Генерация всех ядер согласно конфигурациям.

Возвращает

Dict[str, np.ndarray]: Словарь {имя_ядра: массив_ядра}.

run(save: bool = True, visualize: bool = True) → Dict[str, ndarray][исходный код]

Запуск генерации ядер.

Параметры

savebool: Сохранять ли ядра в файлы.
visualizebool: Показывать ли визуализацию.

Возвращает

Dict[str, np.ndarray]: Сгенерированные ядра.

save(kernels: Dict[str, ndarray]) → None[исходный код]

Сохранение ядер в файлы.

Параметры

kernelsDict[str, np.ndarray]: Словарь ядер для сохранения.

visualize(kernels: Dict[str, ndarray]) → None[исходный код]

Визуализация ядер.

Параметры

kernelsDict[str, np.ndarray]: Словарь ядер для визуализации.