如何使用我自己的数据集训练卷积神经网络以进行图像分类并将其保存为.pb.tensorflow和python中的所有内容

时间:2018-10-20 17:39:09

标签: python tensorflow neural-network

我想学习如何在tensorflow中训练卷积神经网络。可以用一些代码编写一个链接。例如github项目。用我自己的数据集训练卷积神经网络后,如何将其另存为。 pb或.pbtxt。您可以说明执行此操作的过程吗?

1 个答案:

答案 0 :(得分:1)

不确定是否要了解代码还是只是模型。如果只需要模型,请查看此retraining code of tensorflow。为了全面理解,此link可能有用。您引用的pb文件也称为保存的模型,请参见此link