公开(公告)号：CN116612100A

公开(公告)日：2023-08-18

申请号：CN202310638618.X

申请日：2023-05-31

Applicant: 哈尔滨理工大学

Inventor： 李东洁 ,  武学鹏 ,  梁雨 ,  王雪莹 ,  徐东昊 ,  张宇 ,  杨柳

IPC: G06T7/00 ,  G06V10/774 ,  G06V10/40 ,  G06V10/80 ,  G06V10/82 ,  G06N3/045

Abstract: 一种基于改进YOLOv7模型的木材缺陷检测方法。目的是为了提高木材缺陷检测的精度来减少木材加工过程中的浪费。该方法包含以下步骤：步骤1，制作并划分木材缺陷数据集；步骤2，将数据集图像输入骨干特征提取网络中进行特征提取，提取出三个有效特征层；步骤3，将步骤2处理后的特征层输入改进后的FPN网络进行特征融合，输出三个不同尺度的特征层；步骤4，将步骤3中的三个特征层输入预测网络中得到目标的类别信息和位置信息，并使用α‑EIoU损失函数作为损失函数进行训练。该算法可以提高木材缺陷检测的精度，为工厂木材加工提供依据。