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('训练结束')