一种基于Transformer的井下低光照图像增强方法

    公开(公告)号:CN116152117A

    公开(公告)日:2023-05-23

    申请号:CN202310412713.8

    申请日:2023-04-18

    IPC分类号: G06T5/00

    摘要: 本发明公开一种基于Transformer的井下低光照图像增强方法,属于计算机视觉领域。其中,方法包括:首先,获取煤矿井下低光照RGB格式的图像I,将该图像I的格式转化为rawRGB,得到图像Iraw;然后,搭建基于MobileViT模块的两个分支,并采用MobileViT模块的两个分支,基于图像Iraw,预测乘法图M与加法图A;通过Cross Attention模块产生3×3的最终色彩矩阵和参数;最后,根据图像Iraw、乘法图M、加法图A、最终色彩矩阵和参数,建立图像增强网络模型,以得到煤矿井下增强图像,由此,可实现井下低光照图像整体亮度增强的效果,保持图像整体色彩以及保留图像细节信息的目的。

    一种基于Transformer的井下低光照图像增强方法

    公开(公告)号:CN116152117B

    公开(公告)日:2023-07-21

    申请号:CN202310412713.8

    申请日:2023-04-18

    IPC分类号: G06T5/00

    摘要: 本发明公开一种基于Transformer的井下低光照图像增强方法,属于计算机视觉领域。其中,方法包括:首先,获取煤矿井下低光照RGB格式的图像I,将该图像I的格式转化为rawRGB,得到图像Iraw;然后,搭建基于MobileViT模块的两个分支,并采用MobileViT模块的两个分支,基于图像Iraw,预测乘法图M与加法图A;通过Cross Attention模块产生3×3的最终色彩矩阵和参数;最后,根据图像Iraw、乘法图M、加法图A、最终色彩矩阵和参数,建立图像增强网络模型,以得到煤矿井下增强图像,由此,可实现井下低光照图像整体亮度增强的效果,保持图像整体色彩以及保留图像细节信息的目的。