keras加载预训练模型

1.搭建自己的模型：

model = get_net()

2.编译模型：

 model.compile(optimizer=Adam(lr=1e-5), loss=my_loss, metrics=[dice_coef]) 

3.加载预训练模型：

model.load_weights( "my_net.hdf5" )
 

4.如果想继续训练模型：

model.fit_generator(generator=DataGenerator(), epochs=self.epochs,
                            callbacks=[ TensorBoard(log_dir='./log/')])

5.如果只是想使用模型预测结果：

pred=model.predict(test_data, batch_size=2, verbose=1)