扫二维码与项目经理沟通
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
创新互联www.cdcxhl.cn八线动态BGP香港云服务器提供商，新人活动买多久送多久，划算不套路！
小编给大家分享一下Keras模型转TensorFlow的案例，希望大家阅读完这篇文章后大所收获，下面让我们一起去探讨吧！
环境：python 3.6 +opencv3+Keras
训练集：MNIST
下面划重点：因为MNIST使用的是黑底白字的图片，所以你自己手写数字的时候一定要注意把得到的图片也改成黑底白字的，否则会识别错（至少我得到的结论是这样的 ，之前用白底黑字的图总是识别出错）
注意：需要测试图片需要为与训练模时相同大小的图片，RGB图像需转为gray
代码：
import cv2
import numpy as np
from keras.models import load_model
model = load_model('fm_cnn_BN.h6') #选取自己的.h模型名称
image = cv2.imread('6_b.png')
img = cv2.cvtColor(image,cv2.COLOR_RGB2GRAY) # RGB图像转为gray
#需要用reshape定义出例子的个数，图片的 通道数，图片的长与宽。具体的参加keras文档
img = (img.reshape(1, 1, 28, 28)).astype('int32')/255
predict = model.predict_classes(img)
print ('识别为：')
print (predict)
cv2.imshow("Image1", image)
cv2.waitKey(0) 
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
微信二维码
移动版官网