一种适用于小样本微小信号灯的状态识别系统

本发明公开了一种适用于小样本微小信号灯的状态识别系统，包括大样本模型预训练模块、优化冻结策略生成模块、目标域模型训练模块、颜色分类器训练模块、图像采集模块、多阶段小目标检测模块、信号灯状态辅助识别模块和决策模块；优化冻结策略生成模块根据模型在目标检测任务中的性能指标动态选择合适的层冻结组合，在帮助模型适应目标域任务的同时提高训练效率；多阶段小目标检测模块，辅助远距离微小目标的精准识别，解决远距成像不明显导致的颜色识别困难的问题及小尺寸目标成像模糊导致的漏检问题；本发明不仅提高了小样本目标域的泛化性和适应性，而且提升了在复杂环境中对小目标信号灯状态的准确识别能力。