def run(self):
user_num = len(os.listdir('./data/'))
dataset = Dataset('./data/')
#self.pushButton_2.setText('训练中')
dataset.load()
model1 = Model()
model1.build_model(dataset, nb_classes=user_num)
# 先前添加的测试build_model()函数的代码
model1.build_model(dataset, nb_classes=user_num)
# 测试训练函数的代码
model1.train(dataset)
model1.save_model(file_path='./model/aggregate.face.model.h5')
def train_click(self):
#if __name__ == '__main__':
user_num = len(os.listdir('./data/'))
dataset = Dataset('./data/')
self.btn2.setText('训练中')
dataset.load()
model = Model()
model.build_model(dataset, nb_classes=user_num)
# 先前添加的测试build_model()函数的代码
model.build_model(dataset, nb_classes=user_num)
# 测试训练函数的代码
model.train(dataset)
model.save_model(file_path='./model/aggregate.face.model.h5')
self.btn2.setText('训练结束')
