Yolov5移植树莓派实现目标检测

文章预览

一、任务描述  通过机器视觉的方式实现对工件的识别，以树莓派作为上位机，根据不同的识别效果对下位机发出不同指令，控制整个机器的运转，流程如下： 二、数据集获取 这是老师发的零件，所以网上肯定没有直接能用的模型，不像口罩检测等有现成的数据集，所以我们需要自己去获取数据集。 由于我们的零件只有两个，如果我们想获取大量图片话，我们可以通过旋转的方式，我设置每张图片每旋转10度生成一张新的照片，这样没拍一张照片就可以生成36张照片，扩大了数据集的来源。 之后用labelImg对图片进行标注，这样我们就获取了大量的数据集和对应的标签txt文件，我按照test：train：val = 1:8:2的比例分配图片。这样我们的数据集就准备好了。 三、Yolov5模型训练 由于yolov5创作者将模型代码开源，所有我们可以拿来直接用，只需要修改train和 ………………………………