计算机科学
水下
采样(信号处理)
降噪
分段
重要性抽样
概率逻辑
算法
人工智能
计算机视觉
数学
统计
蒙特卡罗方法
数学分析
海洋学
滤波器(信号处理)
地质学
作者
Yi Tang,Hiroshi Kawasaki,Takafumi Iwaguchi
标识
DOI:10.1145/3581783.3612378
摘要
In this paper, we present an approach to image enhancement with diffusion model in underwater scenes. Our method adapts conditional denoising diffusion probabilistic models to generate the corresponding enhanced images by using the underwater images and the Gaussian noise as the inputs. Additionally, in order to improve the efficiency of the reverse process in the diffusion model, we adopt two different ways. We firstly propose a lightweight transformer-based denoising network, which can effectively promote the time of network forward per iteration. On the other hand, we introduce a skip sampling strategy to reduce the number of iterations. Besides, based on the skip sampling strategy, we propose two different non-uniform sampling methods for the sequence of the time step, namely piecewise sampling and searching with the evolutionary algorithm. Both of them are effective and can further improve performance by using the same steps against the previous uniform sampling. In the end, we conduct a relative evaluation of the widely used underwater enhancement datasets between the recent state-of-the-art methods and the proposed approach. The experimental results prove that our approach can achieve both competitive performance and high efficiency. Our code is available at https://github.com/piggy2009/DM_underwater.
科研通智能强力驱动
Strongly Powered by AbleSci AI