一种基于transformer的安全帽检测系统及方法

本发明提供一种基于transformer的安全帽检测系统及方法，其中系统包括：模型训练模块，用于基于迁移学习，利用安全帽检测数据集，训练transformer模型，获得安全帽检测模型；图像获取模块，用于获取施工场景的待检测图像；安全帽检测模块，用于基于所述安全帽检测模型，对所述待检测图像进行安全帽检测。本发明的基于transformer的安全帽检测系统及方法，极大程度上提升了安全帽检测的精准性。