创建了yolov5的虚拟环境
在环境下进行了pytorch的安装(CPU版本,能用但非常慢)
在github上下载了yolov5源码
对ruquirement.txt文件进行了一些修改
并在源码路径下下载requirement中的依赖
完成测试,环境搭建完成
模型检测
关键参数
其他参数
数据集构建
使用opencv对视频进行抽帧处理得到图片
下载labelimg对得到的照片打标签
数据集训练
训练时遇到爆RAM情况,减小batch-size参数(一次看完的图片数),问题解决
又遇到问题
重新训练coco128(训练中)
打开tensorboard 查看各项指标的变化
数据集训练
训练时遇到爆RAM情况,减小batch-size参数(一次看完的图片数),问题解决
又遇到问题
训练集的序号要从0开始,coco训练集正常训练完成
打开tensorboard 查看各项指标的变化
Pyside可视化界面
在虚拟环境下下载
Vscode 下载拓展QT for Python
打开Designer开始设计界面
显示ui界面
完成逻辑设计,可以运行图片/视频检测
———————————————————持续更新———————————————————