VGGFace python 抽取人脸特征

最新推荐文章于 2024-06-21 15:23:03 发布

开心点点

最新推荐文章于 2024-06-21 15:23:03 发布

阅读量2.1k

点赞数

分类专栏： OpenCV_机器视觉

本文链接：https://blog.csdn.net/honghu549599AAA/article/details/78752249

版权

这段代码演示了如何利用LightCNN类以及VGG_FACE模型从图像中抽取人脸特征。通过遍历不同比例的图像，计算特征并输出，然后计算两个不同图像人脸特征之间的余弦相似度，展示了人脸识别的过程。

摘要由CSDN通过智能技术生成

“`

import numpy,cv2
import os,caffe
import sklearn.metrics.pairwise as pw
import time,skimage
import matplotlib.pyplot as plt
import numpy as np
class LightCNN():
def init(self, end_cnn=”eltwise_fc1”, model_version=”LightenedCNN_B”):
self.net = caffe.Net(“VGG_FACE_deploy.prototxt”,”VGG_FACE.caffemodel”, caffe.TEST) #load net
self.end_cnn=”fc7”
self.model_version = model_version

def getFeat(self, imgPath):
    '''
    img = caffe.io.load_image(imgPath,color=False)

    img = skimage.img_as_float(skimage.io.imread(imgPath, as_grey=not False)).astype(np.float32)
    if img.ndim == 2:
      print "img.ndim =sss= 2"
      img = img[:, :, np.newaxis]
    #img = cv2.imread(imgPath,0)
    print "img.ndim===<<><<" + str(img.ndim)
    img = cv2.resize(img,(12