基于GPU优化的绝缘子缺陷检测方法、系统、产品及设备

本发明公开了基于GPU优化的绝缘子缺陷检测方法、系统、产品及设备，涉及计算机视觉技术领域，检测方法包括构建绝缘子缺陷数据集、对网络模型进行改进、降低网络模型的参数量、优化目标检测算法的推理速度以及对绝缘子图像进行实时缺陷检测；检测系统包括数据集构建模块、模型优化模块、算法优化模块和绝缘子检测模块。本发明方法以YOLOX‑tiny模型为基础，设计了基于GPU的模型推理加速方案，通过采用TVM框架中的AutoTVM与Ansor方法，并结合TensorRT推理引擎，在GPU设备上进行部署，以优化模型在GPU上的推理速度，提升网络对正常区域与缺陷区域的区分能力。