“`
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