本文整理匯總了Python中export_inference_graph.main方法的典型用法代碼示例。如果您正苦於以下問題:Python export_inference_graph.main方法的具體用法?Python export_inference_graph.main怎麽用?Python export_inference_graph.main使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類export_inference_graph
的用法示例。
在下文中一共展示了export_inference_graph.main方法的1個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: testExportInferenceGraph
# 需要導入模塊: import export_inference_graph [as 別名]
# 或者: from export_inference_graph import main [as 別名]
def testExportInferenceGraph(self):
tmpdir = self.get_temp_dir()
output_file = os.path.join(tmpdir, 'inception_v3.pb')
flags = tf.app.flags.FLAGS
flags.output_file = output_file
flags.model_name = 'inception_v3'
flags.dataset_dir = tmpdir
export_inference_graph.main(None)
self.assertTrue(gfile.Exists(output_file))