ND4j大矩阵做相识度计算实现1:N计算

现在很多做人脸识别的技术都是基于C++,python;但是java也是有自己的深度学习的库,Deeplearning4j就是java自己的深度学习库,ND4j就是dp4j里面专门做向量计算的库,现在就用ND4j实现1:N矩阵计算功能:

pom:

       <dependency>
            <groupId>org.nd4j</groupId>
            <artifactId>nd4j-native</artifactId>
            <version>${dp4j-version}</version>
        </dependency>

1:N计算的核心就是,用矩阵和一个大矩阵做内积乘法

业务功能:就是用一张人脸的特征和数据库里面有的特征做内积乘法,快速在数据库里面找到这张人脸对应的人;

现实方法:

由于每次都在数据库里面去查询特征数据会消耗时间,所以我把特征数据初始化加载到内存里面来维护;做完内积计算以后需要找到相识度最大的对应的那条数据,opencv里面有一个sortIndex函数可以将数组的值排序以后返回对应的索引地址,java可以自己实现一个排序;

将查出来的特征值进行缓存:

@Component
public class NDCache {


    public static List<BasisFeature> basisFeatures = Lists.newArrayList();

   
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值