Jun, 2024
EndoUIC: 可提示的扩散变压器在胶囊内窥镜中的统一照明校正
EndoUIC: Promptable Diffusion Transformer for Unified Illumination
Correction in Capsule Endoscopy
TL;DRWireless Capsule Endoscopy (WCE) 采用一种无创痛苦的方法被高度重视，尽管硬件限制和复杂的内部动态导致了不均匀的照明，从而导致过曝或欠曝图像。我们引入了EndoUIC，一种使用端到端可提示的扩散变换器（DFT）模型的WCE统一照明校正解决方案，以解决这个问题。我们的工作中，照明提示模块将导航模型以适应不同的曝光水平并执行有针对性的图像增强，在其中自适应提示集成（API）和全局提示扫描器（GPS）模块将进一步增强提示参数和特征之间的并行表示学习。此外，U形恢复DFT模型将捕捉统一照明恢复的长程依赖和上下文信息。此外，我们提供了一个新颖的胶囊内镜暴露校正（CEC）数据集，包括由专业摄影师注释的地面真实图像和损坏图像对。对四个数据集上的各种先进方法进行的大量实验证明了我们提出的方法和组件在WCE照明恢复中的有效性，额外的下游实验进一步展示了其在临床诊断和手术辅助中的实用性。