120狗脸识别检测

本期给大家介绍的是120狗脸识别检测,效果图如下:

这是调用训练好的模型来识别的,运行python 03pyqt.py的可视化界面,通过传狗脸图片来进行识别是哪只狗。

代码仓库和视频演示地址:

120狗脸识别检测_哔哩哔哩_bilibili

代码整体是非常简便的,主要是三个py部分和一个数据集在data文件夹下。

Data文件夹存的500只狗的数据集,每个文件夹对于一只狗的图片集,文件夹名称即为狗的名称,可自行修改狗的名称,也可以替换成其他数据集,只有每个名称为文件夹名,然后每个个体的不同状态下的脸部照片放在该文件夹下即可训练自己的数据集了。

运行01数据集文本生成制作.py可以将上面每只狗的图片路径和对应的标签保存在logs文件夹下的txt文件中,

运行python 02train.py会将txt文本里面的图片路径和对应的标签读取进行训练,训练好的模型也保存在logs文件夹下。这里提供了11种模型可自由的切换使用,

然后训练完成之后还有评价指标图,包括f1-score、精确度和召回率,最后的准确率都是接近1的。

最后运行03pyqt.py会有可视化界面,来加载图片,调用训练好的模型进行识别是哪只狗。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值