{"id":109,"date":"2019-11-27T17:15:19","date_gmt":"2019-11-27T09:15:19","guid":{"rendered":"http:\/\/8.154.33.202\/?p=109"},"modified":"2019-11-27T17:15:19","modified_gmt":"2019-11-27T09:15:19","slug":"tensorflow-%e5%b0%86ckpt%e6%96%87%e4%bb%b6%e5%af%bc%e5%87%ba%e4%b8%bapb%e6%96%87%e4%bb%b6","status":"publish","type":"post","link":"http:\/\/8.154.33.202\/?p=109","title":{"rendered":"tensorflow \u5c06ckpt\u6587\u4ef6\u5bfc\u51fa\u4e3apb\u6587\u4ef6"},"content":{"rendered":"\n<p>tensorflow\u8bad\u7ec3\u65f6\u5c06\u6a21\u578b\u4fdd\u5b58\u4e3ackpt\u6587\u4ef6\uff0c\u5b83\u5305\u542b\u4e86\u7f51\u7edc\u7ed3\u6784\u3001\u7f51\u7edc\u6743\u91cd\u3001\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u95f4\u53d8\u91cf\u7b49\u7b49\u4fe1\u606f\u3002\u800c\u7f51\u7edc\u90e8\u7f72\u4e00\u822c\u662f\u4f7f\u7528pb\u6587\u4ef6\uff0c\u5b83\u5c06\u53d8\u91cf\u4fdd\u5b58\u4e3a\u5e38\u91cf\uff0c\u4ee5\u53ca\u7f51\u7edc\u524d\u5411\u4f20\u64ad\u7684\u6240\u6709\u5fc5\u8981\u7ed3\u6784\u3002\u5982\u4f55\u5c06ckpt\u6587\u4ef6\u5bfc\u51fa\u4e3apb\u6587\u4ef6\uff1f<\/p>\n\n\n\n<p>\u9996\u5148\uff0c\u4f7f\u7528tfrecord\u8bad\u7ec3\u7684ckpt\u4e00\u822c\u5305\u542b\u8bfb\u53d6\u8bad\u7ec3tfrecord\u6587\u4ef6\u7684\u7ed3\u6784\uff0c\u800c\u8fd9\u662fpb\u6587\u4ef6\u6240\u4e0d\u9700\u8981\u7684\u3002pb\u6587\u4ef6\u901a\u5e38\u4f7f\u7528placeholder\u63a5\u53d7\u8f93\u5165\u3002\u56e0\u6b64\uff0c\u8981\u4ee5placeholder\u4e3a\u8f93\u5165\u91cd\u65b0\u5b9a\u4e49\u4e00\u904d\u7f51\u7edc\u7ed3\u6784\uff08\u901a\u5e38\u5c31\u662f\u8c03\u7528\u4e00\u6b21\u7f51\u7edc\u6784\u5efa\u51fd\u6570\uff09\u3002\u5047\u8bbe\u4e3a<br>output = xxnet(input_placeholder)<br>\u8981\u83b7\u53d6\u8f93\u51fa\u8282\u70b9\u7684\u540d\u79f0<br>output_nd_name = output.op.name<\/p>\n\n\n\n<p>\u7136\u540e\uff0c\u8f7d\u5165ckpt\u7684\u6743\u91cd<br>saver = tf.train.Saver()<br>saver.restore(sess, &#8220;xxnet.ckpt&#8221;)<\/p>\n\n\n\n<p>\u7136\u540e\uff0c\u5c06\u5176\u4e2d\u7684\u53d8\u91cf\u8f6c\u5316\u4e3a\u5e38\u91cf\uff0c\u4fdd\u5b58\u6a21\u578b<\/p>\n\n\n\n<pre class=\"wp-block-preformatted\">out_graph_def = tf.graph_util.convert_variables_to_constants(\n    sess=sess,\n    input_graph_def=sess.graph_def,\n    output_node_names=[output_nd_name]\n)\n<strong>with <\/strong>tf.gfile.GFile(<strong>\"xxnet.pb\"<\/strong>,<strong>\"wb\"<\/strong>) <strong>as <\/strong>f:\n    f.write(out_graph_def.SerializeToString())<\/pre>\n","protected":false},"excerpt":{"rendered":"<p>tensorflow\u8bad\u7ec3\u65f6\u5c06\u6a21\u578b\u4fdd\u5b58\u4e3ackpt\u6587\u4ef6\uff0c\u5b83\u5305\u542b\u4e86\u7f51\u7edc\u7ed3\u6784\u3001\u7f51\u7edc\u6743\u91cd\u3001\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u95f4\u53d8\u91cf\u7b49\u7b49\u4fe1\u606f\u3002\u800c\u7f51 &hellip; <\/p>\n<p class=\"link-more\"><a href=\"http:\/\/8.154.33.202\/?p=109\" class=\"more-link\">\u7ee7\u7eed\u9605\u8bfb<span class=\"screen-reader-text\">\u201ctensorflow \u5c06ckpt\u6587\u4ef6\u5bfc\u51fa\u4e3apb\u6587\u4ef6\u201d<\/span><\/a><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[5,4],"tags":[],"class_list":["post-109","post","type-post","status-publish","format-standard","hentry","category-tensorflow","category-4"],"_links":{"self":[{"href":"http:\/\/8.154.33.202\/index.php?rest_route=\/wp\/v2\/posts\/109"}],"collection":[{"href":"http:\/\/8.154.33.202\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/8.154.33.202\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/8.154.33.202\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/8.154.33.202\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=109"}],"version-history":[{"count":1,"href":"http:\/\/8.154.33.202\/index.php?rest_route=\/wp\/v2\/posts\/109\/revisions"}],"predecessor-version":[{"id":110,"href":"http:\/\/8.154.33.202\/index.php?rest_route=\/wp\/v2\/posts\/109\/revisions\/110"}],"wp:attachment":[{"href":"http:\/\/8.154.33.202\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=109"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/8.154.33.202\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=109"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/8.154.33.202\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=109"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}