基于半监督多任务检测的木地板颜色分类及缺陷检测方法
摘要:
本发明涉及一种基于半监督多任务检测的木地板颜色分类及缺陷检测方法,属于图像处理领域。该方法包括:对输入图像进行缺陷色彩校正;标记少量图像作为数据集输入Color‑RCNN模型训练,实现颜色分类和缺陷检测;训练半监督预测监视网络判别伪标签是否准确的;用Color‑RCNN模型预测剩余的未标记图像,实现颜色分类,生成伪标签,用半监督预测监视网络对伪标签进行判别,将正确的检测框留下,过滤掉错误的检测框;将识别正确的图像加入数据集,重新输入Color‑RCNN模型训练,预测剩余的未标记图像,直至未标记图像全部获得正确标签,实现半监督学习。本发明在图像识别的过程中,通过半监督学习,使用少量人工标记和大量未标记的木地板图像训练出模型,提高了分选的效率。
0/0