很值得套用的keras经典Unet版本
代码如下: 1 from model import * 2 from data import *#导入这两个文件中的所有函数 3 4 #os.environ[“CUDA_VISIBLE_DEVICES”] = “0” 5 6 7 data_gen_args = dict(rotation_range=0.2, 8 width_shift_range=0.05, 9 height_shift_range=0.05, 10 shear_range=0.05, 11 zoom_range=0.05, 12 horizontal_flip=True, 13 fill_mode=‘nearest’)#数据增强时的变换方式的字典 14 myGene = trainGenerator(2,‘data/membrane/train’,‘image’,‘label’,data_gen_args,save_to_dir = None) 15 #得到一个生成器,以batch=2的速率无限生成增强后的数据 16 17 model = unet() 18 model_checkpoint = ModelCheckpoint(‘unet_membrane.hdf5’, monitor=‘loss’,verbose=1, save_best_only=True) 19 #回调函数,第一个是保存模型路径