水稻检测模型的训练方法、计数方法、系统、设备和介质

    公开(公告)号:CN117292263A

    公开(公告)日:2023-12-26

    申请号:CN202311329612.0

    申请日:2023-10-13

    Abstract: 本发明提供水稻检测模型的训练方法、计数方法、系统、设备和介质,该训练方法包括:获取水稻籽粒的样本图像和样本图像中水稻籽粒数量的真实值;将样本图像输入至水稻籽粒检测模型中,根据多头自注意力模块和坐标注意力模块,提取样本图像中不同尺度的籽粒特征图;其中,水稻籽粒检测模型以YOLOV5为主网络,并融合多头自注意力模块与坐标注意力模块;根据多头自注意力模块将不同尺度的籽粒特征图进行多尺度特征融合;根据融合后特征图获取样本图像中水稻籽粒数量的预测值;根据真实值、预测值、以及损失函数,调整水稻籽粒模型的参数,获得训练好的水稻籽粒检测模型。可以准确进行千粒重的测定,并且更加高效,精确度更高。

Patent Agency Ranking