blinddeconv.scripts.kernel_generator module

Модуль генерации ядер размытия (PSF) для экспериментов.

Авторы: Куропатов К.Л.

class blinddeconv.scripts.kernel_generator.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]: Словарь ядер для визуализации.

blinddeconv.scripts.kernel_generator.main() → None[исходный код]