一种基于对比学习的GIS设备隔离刀闸动作识别方法

本发明提供了一种基于对比学习的GIS设备隔离刀闸动作识别方法，其包括以下具体步骤：采用摄像头获取不同GIS设备的开关状态的视频；提取不同GIS设备的开关状态的视频的图片；标注提取的图片；构建GIS设备的数据集图片，包含训练集图片和测试集图片；建立GIS设备的特征提取模型并进行模型训练；将标注好的图片加载到GIS设备的特征提取模型的vision transformer和tripletloss网络中训练，得到模型最佳权重数据；将最佳权重数据加载到vision transformer和tripletloss网络中，输入测试集图片，输出刀闸开关状态，完成GIS设备隔离刀闸动作识别。本申请通过基于邻注意力机制的vision transformer网络提取图片特征并通过tripletloss增加不同类别图片的特征差异度。提高准确率，有效地减少了人力和时间成本。