基于PLSA-VQGAN的排水管道图像修复方法

本发明涉及的是一种基于PLSA‑VQGAN的排水管道图像修复方法，其中，方法包括：在实验管道模型里用双目相机采集管道图像，利用基于CLIPSEG的绳索分割算法和基于可变形卷积的机器人连接件分割算法得到的管道图像掩码与管道图像，并划分训练数据集和测试数据集；基于PLSA‑VQGAN网络结构，在码本学习阶段，得到一个量化码本以及可以进行图像特征提取和重建的解码器编码器；在内容推理阶段，使用基于Transformer架构的内容推理网络构建管道遮挡图像修复网络。将训练数据集送入基于Transformer架构的内容推理网络，训练神经网络模型，用测试数据集测试训练好的管道遮挡图像修复网络模型，实现管道图像的修复，并对修复效果进行客观评价。本发明能有效实现地下管道视频图像的修复，并对图像的细节方面进行增强，可用于排水管道检测机器人的视频监控、地下管道病害智能检测识别等科学领域。