def evaluate1_lbp():
filters = Gabor().build_filters()
from tqdm import tqdm
from data import CK, Fer2013, Jaffe
_, x, y = Fer2013().gen_train_no()
train = []
for i in tqdm(np.arange(0, x.shape[0], 1)):
res = LBP().get_lbp(x[i])
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
train.append(res)
train = np.array(train)
test = []
_, x, y = Jaffe().gen_train_no()
for i in tqdm(np.arange(0, x.shape[0], 1)):
res = LBP().get_lbp(x[i])
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
test.append(res)
test = np.array(train)
Classifier().SVM(train, test)
test = []
_, x, y = CK().gen_train_no()
for i in tqdm(np.arange(0, x.shape[0], 1)):
x[i] = preliminary_treatment.gray_norm(x[i])
x[i] = preliminary_treatment.adaptive_histogram_equalization(x[i])
res = Gabor().getGabor(x[i], filters, False, 6)
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
test.append(res)
test = np.array(train)
Classifier().SVM(train, test)
def evaluate_valid(data_op=1, op=1, reduction=1, rate=0.2):
"""
评估函数，在三个数据集上进行评估
:param op: 1-all 2-part
:param data_op: 1-CK 2-Fer 3-Jaffe
:return:
"""
import preprocess
from tqdm import tqdm
from data import CK, Fer2013, Jaffe
filters = Gabor().build_filters()
if data_op == 1:
_, x, y = CK().gen_train_no()
if data_op == 2:
_, x, y = Fer2013().gen_train_no()
if data_op == 3:
_, x, y = Jaffe().gen_train_no()
train = []
if op == 1:
for i in tqdm(np.arange(0, x.shape[0], 1)):
x[i] = preprocess.gray_norm(x[i])
x[i] = preprocess.adaptive_histogram_equalization(x[i])
res = Gabor().getGabor(x[i], filters, False, reduction)
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
train.append(res)
train = np.array(train)
if data_op != 2:
# 需要划分
from sklearn.model_selection import train_test_split
x_train, x_test, y_train, y_test = train_test_split(train, train, random_state=2019, test_size=rate)
Classifier().SVM(x_train, x_test)
test1 = []
test2 = []
if data_op == 2:
_, x, y = Fer2013().gen_valid_no(1)
for i in tqdm(np.arange(0, x.shape[0], 1)):
x[i] = preprocess.gray_norm(x[i])
x[i] = preprocess.adaptive_histogram_equalization(x[i])
res = Gabor().getGabor(x[i], filters, False, reduction)
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
test1.append(res)
_, x, y = Fer2013().gen_valid_no(2)
for i in tqdm(np.arange(0, x.shape[0], 1)):
x[i] = preprocess.gray_norm(x[i])
x[i] = preprocess.adaptive_histogram_equalization(x[i])
res = Gabor().getGabor(x[i], filters, False, reduction)
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
test2.append(res)
test1 = np.array(test1)
test2 = np.array(test2)
print("Public")
Classifier().SVM(train, test1)
print("Pirvate")
Classifier().SVM(train, test2)
def evaluate_test():
"""
在未训练的数据集上进行测试
:return:
"""
filters = Gabor().build_filters()
from tqdm import tqdm
from data import CK, Fer2013, Jaffe
_, x, y = Fer2013().gen_train_no()
train = []
for i in tqdm(np.arange(0, x.shape[0], 1)):
x[i] = preprocess.gray_norm(x[i])
x[i] = preprocess.adaptive_histogram_equalization(x[i])
res = Gabor().getGabor(x[i], filters, False, 6)
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
train.append(res)
train = np.array(train)
test = []
_, x, y = Jaffe().gen_train_no()
for i in tqdm(np.arange(0, x.shape[0], 1)):
x[i] = preprocess.gray_norm(x[i])
x[i] = preprocess.adaptive_histogram_equalization(x[i])
res = Gabor().getGabor(x[i], filters, False, 6)
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
test.append(res)
test = np.array(train)
Classifier().SVM(train, test)
test = []
_, x, y = CK().gen_train_no()
for i in tqdm(np.arange(0, x.shape[0], 1)):
x[i] = preprocess.gray_norm(x[i])
x[i] = preprocess.adaptive_histogram_equalization(x[i])
res = Gabor().getGabor(x[i], filters, False, 6)
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
test.append(res)
test = np.array(train)
Classifier().SVM(train, test)
ModelCheckpoint('../models/cnn3_best_weights.h5',
monitor='val_acc',
verbose=True,
save_best_only=True,
save_weights_only=True)
]
history_jaffe = model.fit_generator(
train_generator,
steps_per_epoch=len(y_train) // opt.batch_size,
epochs=opt.epochs,
validation_data=valid_generator,
validation_steps=len(y_valid) // opt.batch_size,
callbacks=callback)
his = history_jaffe
else:
expr, x, y = CK().gen_train()
y = to_categorical(y).reshape(y.shape[0], -1)
# 划分训练集验证集
x_train, x_valid, y_train, y_valid = train_test_split(x,
y,
test_size=0.2,
random_state=2019)
print(
"load CK+ dataset successfully, it has {} train images and {} valid iamges"
.format(y_train.shape[0], y_valid.shape[0]))
train_generator = ImageDataGenerator(rotation_range=10,
width_shift_range=0.05,
height_shift_range=0.05,
horizontal_flip=True,
shear_range=0.2,
zoom_range=0.2).flow(
def evaluate_lbp(data_op=1, op=1, reduction=1, rate=0.2):
filters = Gabor().build_filters()
from tqdm import tqdm
from data import CK, Fer2013, Jaffe
_, x, y = CK().gen_train_no()
if data_op == 2:
_, x, y = Fer2013().gen_train_no()
if data_op == 3:
_, x, y = Jaffe().gen_train_no()
train = []
if op == 1:
for i in tqdm(np.arange(0, x.shape[0], 1)):
x[i] = preliminary_treatment.gray_norm(x[i])
x[i] = preliminary_treatment.adaptive_histogram_equalization(x[i])
res = Gabor().getGabor(x[i], filters, False, reduction)
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
train.append(res)
train = np.array(train)
if op == 2:
for i in tqdm(np.arange(0, x.shape[0], 1)):
img, dets, shape_list, img_list, pt_post_list = preliminary_treatment.deal(x[i])
res1 = Gabor().getGabor(img, filters, 0, 1)
res1 = np.array(res1)
res = []
if len(shape_list) == 0:
continue
for _, pt in enumerate(shape_list[0].parts()):
px, py = min(max(pt.x, 0), 47), min(max(pt.y, 0), 47)
im = res1[0]
cv2.circle(im, (px, py), 2, (255, 0, 0), 1)
res.append(res1[:, px, py])
res = np.array(res)
res = np.append(res, y[i])
train.append(res)
train = np.array(train)
if op == 3:
for i in tqdm(np.arange(0, x.shape[0], 1)):
res = LBP().get_lbp(x[i])
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
train.append(res)
train = np.array(train)
if data_op != 2:
from sklearn.model_selection import train_test_split
x_train, x_test, y_train, y_test = train_test_split(train, train, random_state=2019, test_size=rate)
Classifier().SVM(x_train, x_test)
test1 = []
test2 = []
if data_op == 2:
_, x, y = Fer2013().gen_valid_no(1)
for i in tqdm(np.arange(0, x.shape[0], 1)):
res = LBP().get_lbp(x[i])
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
test1.append(res)
_, x, y = Fer2013().gen_valid_no(2)
for i in tqdm(np.arange(0, x.shape[0], 1)):
res = LBP().get_lbp(x[i])
res = np.array(res).reshape(-1)
res = np.append(res, y[i])
test2.append(res)
test1 = np.array(test1)
test2 = np.array(test2)
print("Public")
Classifier().SVM(train, test1)
print("Pirvate")
Classifier().SVM(train, test2)