训练模型部分
https://github.com/ultralytics/yolov5下载yolov5项目
然后正常安装需要的包 pip install -r requirements.txt
在训练模型之前需要先采集下数据集,然后使用标注工具,对数据进行标注。这里采用https://github.com/tzutalin/labelImg这个项目。
下载解压后一定删除labelImg-master\data\predefined_classes.txt
不然后面多出很多标签
标注好之后,按ctrl + s进行保存后生产txt的数据。
图片放images里面,标记数据放labels里面就可以了。
之后,创建yaml文件,配置下训练的数据集位置就ok啦!!!
nc:代表训练类别
names:代表标记的描述
train:训练数据集位置
val:验证数据集位置
最后说下训练需要改的东西:
- 训练模型的选择
- yaml数据集的位置
- epochs 训练的次数
- imgsz训练图片的大小,这个非常重要
校验训练模型部分
- 修改校验模型
- 数据集的位置
- 图片大小
- 开始校验
- 最后提取成一个flask就可以了
过验证码部分
这部分遇到一个坑,selenium 先进行按住滑块拖动的第一次后不能释放鼠标动作!!!!
最后
yolov5参考文章:https://blog.csdn.net/qq_45701791/article/details/113992622