Windows：C++调用python文件调用keras模型推理

最新推荐文章于 2022-11-21 19:15:51 发布

Stairs002

最新推荐文章于 2022-11-21 19:15:51 发布

阅读量855

点赞数 2

分类专栏： Tensorflow&keras 文章标签： python 深度学习 tensorflow c++ windows

本文链接：https://blog.csdn.net/qq_39363145/article/details/108701659

版权

本文介绍了如何在Windows环境中，使用C++程序调用Python脚本，进而执行Keras深度学习模型的推理过程。主要包括三个步骤：首先编写Python文件，然后创建C++文件进行接口调用，最后编译Python运行时所需文件。

摘要由CSDN通过智能技术生成

Step1：Python文件

from tensorflow.keras.models import load_model
import cv2 as cv
import numpy as np

def create_classifier(model_path):
    model = load_model(model_path)
    return model

def model_predict(model,image_path):
    img = cv.imread(image_path)
    img = cv.resize(img, (224,224))
    image = np.expand_dims(img,axis=0)
    result = model.predict(image)
    print(result)
    return result