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