一种基于YOLOv5s神经网络的酒店套装门检测方法
摘要:
本发明公开了一种基于YOLOv5s神经网络的酒店套装门检测方法,属于深度学习技术领域。该方法包括以下步骤:(1)采集酒店安装好的套装门图像并使用ACDSee批量调整图像大小成640×640,(2)通过LabelImg目标检测标注工具对数据进行标记,(3)使用PyTorch深度学习框架,搭建基础的YOLOv5s神经网络模型并修改模型结构,(4)利用数据集对目标检测模型进行训练和测试,得到最优目标检测模型并部署到华为Atlas200DK边端设备上。
0/0