tensorflow读取一个模型后多次使用

最新推荐文章于 2022-12-28 14:21:47 发布

碧影江白

最新推荐文章于 2022-12-28 14:21:47 发布

阅读量2.3k

点赞数 1

分类专栏：深度学习文章标签： python tensorflow

本文链接：https://blog.youkuaiyun.com/qq_34470213/article/details/107128900

版权

训练好一个模型后，将其投入使用，会有在项目初始化后多次加载测试数据的需求，可以采用保存graph的思想实现

（在一个项目中需要加载多个模型同样可用）

另：这条博客接我的上一条https://blog.youkuaiyun.com/qq_34470213/article/details/104076898，是在上一个代码的基础上改写的。

1、新建文件test.py，建一个类Model_test，用来保存模型，包括一个初始化方法，用来初始化模型（项目中仅需初始化时调用一次），一个测试调用方法，用来调用模型进行测试（每次测试调用一次）。

class Model_test():
    def restore(self):
        self.model = Model.LeNet5(1, 5)
        path = "D:/model/model/model.ckpt"
        self.model.load(path)

    def restore_test(self, image_path):
        image = Process.process_one(image_path)
        sort = self.model.test1(image)
        return sort