上QQ阅读APP看书,第一时间看更新
AI源码解读.数字图像处理案例:Python版
5.3.4 模型测试
该测试由两部分组成:一是调用摄像头获取实时图片;二是将图片转换为数据,输入已经训练好的卷积网络模型中,并且获取输出(逻辑判断)。
1.读取模型及调用
相关代码如下:
2.模型导入及调用
定义get_inputs()和cat_detect()函数,调用模型对照片进行有猫与否的判断。对照片的图像进行处理,这里和模型在数据上的操作一致,以保证检测过程中拥有和模型测试阶段一样的高准确率。
3.拍照功能
本部分包括自动和手动拍照,以下代码用于Pycharm中实现拍照功能并进行测试。
1)自动拍照
自动拍照主要依靠OpenCV以及训练好的猫脸检测与识别模型实现。首先,通过OpenCV控制摄像头获取画面;其次,调用已经训练好的模型,对每一帧图像进行识别。如果识别到猫脸,则自动拍照,显示照片并保存在指定位置。
2)手动拍照
手动拍照依靠OpenCV以及训练好的猫脸检测与识别模型实现。首先,通过OpenCV控制摄像头获取画面;其次,调用已经训练好的模型,对每一帧图像进行识别。
4.前端代码
本部分包括图像增广、模型微调、自动拍照、手动拍照、界面设计和打包.exe文件。
1)图像增广
相关代码如下:
2)模型微调
相关代码如下:
3)自动拍照
相关代码如下:
4)手动拍照
相关代码如下:
5)界面设计
相关代码如下:
6)打包.exe文件
相关代码如下:
pyinstaller - F - w cat_photo.py #去除黑窗口打包为.exe文件