本期给大家介绍的是122猫脸识别检测,效果图如下:
本期检测是猫脸识别,如上根据加载猫的图片,识别猫的名字,数据集中,该猫是在0102文件夹下,命名为0102,当然也可以自定义文件夹名称来修改猫的名称。
代码下载和视频演示地址:
整个文件夹主要是data数据集文件夹和3个py文件
其中data文件夹下供有191只猫的图片,每个文件夹为一只猫在不同状态下拍摄的图片。
运行01数据集文本生成制作.py可以将图片路径保存在txt文件里,并附上对应的标签。
运行02train.py可以将图片路径读取训练模型,也保存在爱logs文件夹下。在训练时提供了11种CNN网络可任意切换使用。
最后训练结束会有f1-score、精确率和召回率评价指标图保存在result文件夹下。
运行03pyqt.py会生成可视化界面用于加载猫脸图片识别,看看是哪只猫。