这里分类和汇总了欣宸的全部原创(含配套源码):https://github.com/zq2599/blog_demos
本文是《JavaCV的摄像头实战》系列的第十三篇,前文《JavaCV的摄像头实战之十二:性别检测》中,借助训练好的卷积神经网络模型开发出了识别性别的应用,今天在前文基础上做少量改动,实现年龄识别的功能,效果如下图:
应用主要功能如下图所示:
如果您看过《JavaCV的摄像头实战》系列的其他文章,就会发现上图中只有蓝色部分是新增内容,其余的步骤都是固定套路,《JavaCV的摄像头实战》系列的每一个应用玩的都是相同套路:别看步骤挺多,其实都是同一个流程
名称 | 链接 | 备注 |
---|---|---|
项目主页 | https://github.com/zq2599/blog_demos | 该项目在GitHub上的主页 |
git仓库地址(https) | https://github.com/zq2599/blog_demos.git | 该项目源码的仓库地址,https协议 |
git仓库地址(ssh) | git@github.com:zq2599/blog_demos.git | 该项目源码的仓库地址,ssh协议 |
public static void main(String[] args) {
String base = "E:\\temp\\202112\\25\\opencv\\";
DetectService detectService = new AgeDetectService(
base + "haarcascade_frontalface_alt.xml",
base + "age\\deploy.prototxt",
base + "age\\age_net.caffemodel");
new PreviewCameraWithGenderAge(detectService).action(1000);
}
package com.bolingcavalry.grabpush.extend;
import lombok.extern.slf4j.Slf4j;
import org.bytedeco.javacpp.DoublePointer;
import org.bytedeco.opencv.opencv_core.Mat;
import org.bytedeco.opencv.opencv_core.Point;
import static org.bytedeco.opencv.global.opencv_core.minMaxLoc;
/**
* @author willzhao
* @version 1.0
* @description 检测年龄的服务
* @date 2021/12/3 8:09
*/
@Slf4j
public class AgeDetectService extends GenderDetectService {
/**
* 设置训练模型时划分的年龄段,所以推理结果也是这样的年龄段
*/
private static final String[] AGES = new String[]{"0-2", "4-6", "8-13", "15-20", "25-32", "38-43", "48-53", "60-"};
/**
* 构造方法,在此指定proto和模型文件的下载地址
*
* @param classifierModelFilePath
* @param cnnProtoFilePath
* @param cnnModelFilePath
*/
public AgeDetectService(String classifierModelFilePath, String cnnProtoFilePath, String cnnModelFilePath) {
super(classifierModelFilePath, cnnProtoFilePath, cnnModelFilePath);
}
@Override
protected String getDescriptionFromPredictResult(Mat prob) {
DoublePointer pointer = new DoublePointer(new double[1]);
Point max = new Point();
// 把prob理解为一个数组,
// 第一个元素是"0-2"的置信度
// 第二个元素是"4-6"的置信度
// 第三个元素是"8-13"的置信度
// 第四个元素是"15-20"的置信度
// ...
// 第八个元素是"60-"的置信度
// minMaxLoc方法帮忙我们找出了置信度最高的元素,max是元素位置,pointer是这个元素的置信度
minMaxLoc(prob, null, pointer, null, max, null);
// 如果置信度太低,那就是"难以置信",就返回空字符串
if (pointer.get()<0.6d) {
return "";
} else {
// 如果置信度可信,就返回该元素对应的年龄范围
return AGES[max.x()];
}
}
}