Я пытаюсь обучить модель unet, и моя основная программа выглядит примерно так:
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 Обновите ваш Model
вызов API Keras 2: Model(inputs=Tensor("in..., outputs=Tensor("co...)
модель = Модель (вход = входы, выход = конв10)
Я понимаю, что это потому, что мой model.py является локальным. почему я хотел обновить его и не смог сделать. Не могли бы вы сказать мне, как я могу это сделать?
Еще одна вещь, которую мой код показывает на 70-75% точности. Я хочу улучшить свой код как минимум на 90% для текущей программы, что я должен добавить для лучших результатов?
Спасибо !!