TensorFlow:存储的训练模型在哪里以及如何访问?

时间:2017-07-07 16:42:43

标签: machine-learning tensorflow neural-network caffe

TensorFlow在哪里以及如何存储其模型(图表)?它在public Image<Bgr, Byte> CreateImageFromBytesArray(byte[] bytes) { using (var ms = new MemoryStream(bytes)) { Bitmap bmp = new Bitmap(ms); Image<Bgr, Byte> img = new Image<Bgr, Byte>(bmp); //EXCEPTION HERE return img; } } 个文件中吗?

我知道在Caffe中,模型存储在checkpoint文件中,该文件用于在模型训练后部署(测试)模型。

我刚刚在tutorial中训练了模型,现在我的.caffemodel目录中有以下文件:

files

哪些文件包含实际模型?

例如,如果我想使用这个训练过的模型对单个图像进行分类,那么使用哪个文件进行分类?

1 个答案:

答案 0 :(得分:3)

是的,该模型位于各种checkpoint文件中 - 尽管*.ckpt文件现已拆分为dataindex个文件。我不确定索引文件中的内容,但是 *.data*个文件包含权重和偏差值。 *.meta文件包含图形:变量,数据结构,转换等。

Here是关于护理和喂养的很好的教程。