2020
09-29
09-29
tensorflow转换ckpt为savermodel模型的实现
ckpt转换成SavedModelconvert_ckpt_to_savermodel.pyimporttensorflowastfimportsystrained_checkpoint_prefix=sys.argv[1]export_dir=sys.argv[2]graph=tf.Graph()config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True)withtf.compat.v1.Session(graph=graph,config=config)assess:#Restorefromcheckpointloader=tf.compat.v1.train.import_meta_graph(trained_checkpoint_prefi...
继续阅读 >