本文操作的系统是ubuntu20.04,其他版本可借鉴操作。
1.安装pycharm
上面的链接里面有详细的步骤,一步步照做即可完成pycharm的安装
到此完成pycharm的安装
2.yolo 环境配置
本文使用的是yolov4,tensorflow2框架。首先是yolov4程序代码
权重文件在我的首页资源里面,自己下载,解压到程序里面,最终长这样。
这里的代码是借鉴大神的优化代码
里面有yolov4的详细原理,以及如何训练自己的模型,后期如果想自己开发可以借鉴。
将这些完成后开始在ubuntu上面安装anaconda
2.1 安装anaconda
1.下载安装包
点击链接,下载安装包
2.解压安装包
bash Anaconda3-2023.03-Linux-x86_64.sh
输入上面的命令安装
完成anaconda 的安装
2.2 anaconda安装tensorflow2.2环境
打开终端
# 创建一个虚拟环境
conda create -n tensorflow2 python=3.8
# 激活环境
conda activate tensorflow2
接下来安装各种包
各种包的版本很麻烦,一个搞错就得走很多坑,如果你想自己试试可以尝试,如果你不想走弯路直接按照我这个配置即可
在终端直接pip install
到此完成环境配置
3.训练自己的数据集
https://blog.csdn.net/weixin_44791964/article/details/106533581
里面有怎么样训练自己的数据集
如果是训练自己的数据集,annotation mode 一定要改成 0
最好是去b站看大神的视频跟着配置,有很多细节要注意,在b站直接输yolov4 tensorflow tiny找到原视频。
4.ubuntu安装mysql
跟着链接安装mysql,过程中可能会有一些bug,有问题直接在复制粘贴博客找方法。
5.pycharm导入mysql数据库
5.1 pycharm安装db navigator
点击file,选择settings
在搜索框搜索database,下载database navigator
5.2 连接数据库
在里面填入自己的信息,再点击 右下角的test connection
如果失败,显示不允许访问,将数据库的host改为%
+--------------+-------------------------+
| Host | User |
+--------------+-------------------------+
| % | root |
| localhost | debian-sys-maint |
| localhost | mysql.session |
| localhost | mysql.sys |
+--------------+-------------------------+
4 rows in set (0.00 sec)
显示为这种格式,具体操作如下
use mysql;
update user set host = '%' where user = 'root';
我当时卡在这一下午,一直显示禁止访问,如果解决不了自己去博客上找找报错原因
连接好之后可以看到像左边一样能够看到数据库的数据
6.修改程序,将数据导入数据库
我自己训练的模型是识别人的数量
在运行predict的时候,控制台会打印这些数据,我们通过观察,想要得到的数据就在found {} boxes for img 里面,所以在yolo.py里面找到这个变量,再将它的值传入mysql里面即可。
图中倒数第二行里面我们找到了这个变量,为len(out_boxes),我们先连接数据库。注意sql语句里面的INSERT INTO 后面的my是我自己建的数据库,你要把它改为自己的数据库,后面括号里面的是你自己设置的属性,如果你只是想输出一个类别的数量,直接复刻我的格式就行。
对了,别忘了先在终端下载pymysql,然后在yolo.py引用import pymysql,这是pycharm连接mysql的库。
注意,连接数据库全部都是在yolo.py里面,写完先运行yolo.py 再运行predict.py,这样在你识别的时侯就能一边识别一边传输了。
最后打开数据库,可以看到人数在数据库里面实时更新。
至此全部结束,对于小白来说比较困难,但坚持下去会有很大的收获,有问题评论区见.