В процессе формирования изображения камерой из-за изменения угла обзора возникает эффект виньетирования, при котором изображение светлее в центре и темнее по краям. Наличие виньетирования приводит к потере части текстурной информации по краям изображения, что значительно влияет на производительность обработки машинного зрения. Для решения этой проблемы в данной работе предложены улучшения модели сети Retinex-Net с двух сторон: коррекции четкости изображения и повышения эффективности шумоподавления. Во-первых, в исходную модель добавлены дилатационные свёртки для сохранения высокого разрешения скорректированного изображения и расширения поля восприятия. Во-вторых, шумоподавление усовершенствовано с применением плотной остаточной сети с целью плотного извлечения признаков каждого уровня виньетированного изображения, сохранения деталей изображения и подавления шума. Наконец, создан набор данных виньетированных изображений и проверена производительность предложенного алгоритма на тестовом наборе. По сравнению с исходной моделью значение SSIM увеличилось на 0.293, PSNR — на 0.727, RMSE снизилось на 0.095. По сравнению с алгоритмами коррекции, основанными на минимизации энтропии изображения, адаптивной компенсации Retinex и симметрии радикального градиента, алгоритм данной работы показывает лучшие результаты коррекции и визуально более удобен для наблюдения и понимания.

коррекция виньетирования изображений; теория Retinex; дилатационные свёртки; остаточные сети