一种基于小麦图像的目标检测算法

    公开(公告)号:CN112488006A

    公开(公告)日:2021-03-12

    申请号:CN202011406565.1

    申请日:2020-12-05

    Applicant: 东南大学

    Abstract: 本发明涉及一种基于小麦图像的目标检测算法,所述检测算法包括以下步骤:步骤1:第一阶段为算法的训练阶段,步骤2:第二阶段为算法的实时推理阶段。针对小麦数据集优化后的YOLOv5算法具有较高的精度和较高的推理速度,可以实现实时推理,其线下训练也可在单块GPU上完成(例如GTX1080ti或者RTX2080Ti等),极大的降低了对硬件的要求。

Patent Agency Ranking