学会用YOLO对简单的目标进行检测--Datawhale AI 夏令营

通过在Datawhale Ai 夏令营中的学习学会了用yolo对目标进行检测。

1、读取数据

2、通过对视频流的逐帧处理成图片。

3、通过对坐标的读取，再用matplotlib的工具包对图片进行绘制

4、将下好的数据转换为yolo需要的格式，txt文件中放数据的坐标、类别信息，一个文件放视频流处理完的图片文件，两个文件一一对应。

5、下载需要的模型权重

6、用下好的模型和设置好的yaml文件对训练集进行训练，可以设置自己需要的epoch，和batch-size，看自己设备情况而定。

7、模型跑完后对跑完的模型权重进行保存。