-
公开(公告)号:CN112465794A
公开(公告)日:2021-03-09
申请号:CN202011425522.8
申请日:2020-12-10
Applicant: 无锡卡尔曼导航技术有限公司
Abstract: 本发明涉及图像处理领域和深度学习领域,具体涉及一种基于YOLOv4和嵌入式平台的高尔夫球检测方法。它先制作高尔夫球数据集;接着,在电脑端搭建Darknet框架所需软件环境,嵌入式设备烧录系统镜像必需软件;然后,对数据集的高尔夫球目标框运行K‑means++聚类;接着,使用YOLOv4模型对高尔夫数据集进行训练;之后,将电脑端训练好的YOLOv4权重转化为ONNX格式;之后,将ONNX格式文件转换为TensorRT引擎文件;最后,使用引擎文件对高尔夫球进行识别即可。采用该方法可基本避免漏检的情况,模型部署的难度较小,较Darknet框架运行的速度更快。