我正在尝试培训一个unet模型,我的主要课程是smth,如下所示:
data_gen_args = dict(rotation_range=0.2,
width_shift_range=0.05,
height_shift_range=0.05,
shear_range=0.05,
zoom_range=0.05,
horizontal_flip=True,
fill_mode='nearest')
myGene=trainGenerator(2,'PATH','org','png',data_gen_args,save_to_dir = None)
model = unet()
model_checkpoint = ModelCheckpoint('unet_membrane.hdf5', monitor='loss',verbose=1, save_best_only=True)
model.fit_generator(myGene,steps_per_epoch=100,epochs=10, callbacks=[model_checkpoint])
当它开始学习时,会出现如下错误:
UserWarning: Update your
Model
call to the Keras 2 API:Model(inputs=Tensor("in..., outputs=Tensor("co...)
model = Model(input = inputs, output = conv10)
我明白这是因为我的model.py是本地的,这就是为什么我想更新它,但无法做到的原因。你能告诉我怎么做吗
还有一件事,我的代码显示了70-75%的准确率,我想为当前程序改进代码至少90%,我应该添加什么以获得更好的结果
谢谢
您的模型似乎是使用
Keras 1
syntax创建的:代替
Keras 2
{a2}:只要改变这一点,它就不会抱怨了
关于您的准确性,我们不知道您是如何创建模型的,您的数据是什么,您试图解决什么问题。因此,这取决于任务的类型和您的探索
相关问题 更多 >
编程相关推荐