废话少说直接上代码样例如下importtensorflowastfimportosfromtensorflow.python.toolsimportfreeze_graph#本来这个model本无需解释太多,但是这么多人不能耐下心来看,那么我简单的说一下吧#network是你们自己定义的模型结构而已#ps:#defnetwork(input):#returntf.layers.max_pooling2d(input,2,2)frommodelimportnetworkos.environ['CUDA_VISIBLE_DEVICES']='2'#设置GPUmodel_path="pathto/model.ckpt-...
继续阅读 >