2018-05-09 16:18:38 +08:00
|
|
|
import face_model
|
2018-01-26 13:07:00 +08:00
|
|
|
import argparse
|
|
|
|
|
import cv2
|
2018-05-09 16:18:38 +08:00
|
|
|
import sys
|
2018-01-26 13:07:00 +08:00
|
|
|
import numpy as np
|
|
|
|
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='face model test')
|
|
|
|
|
# general
|
|
|
|
|
parser.add_argument('--image-size', default='112,112', help='')
|
2018-05-18 10:41:21 +08:00
|
|
|
parser.add_argument('--model', default='', help='path to load model.')
|
2018-01-26 13:07:00 +08:00
|
|
|
parser.add_argument('--gpu', default=0, type=int, help='gpu id')
|
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
2020-11-07 13:22:05 +08:00
|
|
|
vec = args.model.split(',')
|
|
|
|
|
model_prefix = vec[0]
|
|
|
|
|
model_epoch = int(vec[1])
|
|
|
|
|
model = face_model.FaceModel(args.gpu, model_prefix, model_epoch)
|
2018-05-18 10:41:21 +08:00
|
|
|
img = cv2.imread('Tom_Hanks_54745.png')
|
2018-05-09 16:18:38 +08:00
|
|
|
img = model.get_input(img)
|
2020-11-07 13:22:05 +08:00
|
|
|
|
|
|
|
|
f1 = model.get_feature(img)
|
2018-01-26 13:07:00 +08:00
|
|
|
f2 = model.get_feature(img)
|
2020-11-07 13:22:05 +08:00
|
|
|
sim = np.dot(f1, f2)
|
|
|
|
|
assert(sim>=0.99 and sim<1.01)
|
|
|
|
|
|
