将增量训练完的.h5模型文件转换为.weights文件

chineseocr / table-detect

table detect(yolo) , table line(unet)

MIT License

236 stars 86 forks source link

将增量训练完的.h5模型文件转换为.weights文件 #10

Open aleassx opened 3 years ago

aleassx commented 3 years ago

如题，如何将增量训练完的.h5模型文件转换为.weights文件

dcdethan commented 3 years ago

train.py 20line checkpointer = ModelCheckpoint(filepath=filepath,monitor='loss',verbose=0,save_weights_only=True, save_best_only=True) 这里默认就是保存权重至于载入load.weights就可以了

niconchan commented 1 year ago

加载的weight存到哪去了呢，没看到train.py里写了weight的存储路径