通过在Datawhale Ai 夏令营中的学习学会了用yolo对目标进行检测。
1、读取数据
2、通过对视频流的逐帧处理成图片。
3、通过对坐标的读取,再用matplotlib的工具包对图片进行绘制
4、将下好的数据转换为yolo需要的格式,txt文件中放数据的坐标、类别信息,一个文件放视频流处理完的图片文件,两个文件一一对应。
5、下载需要的模型权重
6、用下好的模型和设置好的yaml文件对训练集进行训练,可以设置自己需要的epoch,和batch-size,看自己设备情况而定。
7、模型跑完后对跑完的模型权重进行保存。