上期文章我們分享了人工智能Keras圖像分類器(CNN卷積神經網絡的圖片識別的訓練模型)，本期我們使用預訓練模型對圖片進行識別：Keras CNN卷積神經網絡模型訓練

導入第三方庫

from keras.preprocessing.image import img_to_arrayfrom keras.models import load_modelimport numpy as npimport imutilsimport cv2

加載圖片

image = cv2.imread("examples/00000002.jpg") # 加載一個圖片orig = image.copy() # 復制圖片image = cv2.resize(image, (28, 28))# 圖片resizeimage = image.astype("float") / 255.0 #圖片轉換為0 1數據image = img_to_array(image) # arrayimage = np.expand_dims(image, axis=0)# 增加一個軸

加載預訓練模型

model = load_model("lenet.model")

圖片識別

(notsnowman, snowman) = model.predict(image)[0]# 圖片預測# 顯示標簽與預測數據label = "snowman" if snowman > notsnowman else "notsnowman"proba = snowman if snowman > notsnowman else notsnowmanlabel = "{}: {:.2f}%".format(label, proba * 100)

顯示圖片

output = imutils.resize(orig, width=400)cv2.putText(output, label, (10, 25), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 255, 0), 2)cv2.imshow("Output", output)cv2.waitKey(0)

關注微信公眾號：啟示AI科技

可以體驗不一樣的AI工具喲

總結

以上是生活随笔為你收集整理的cnn图像二分类 python_人工智能Keras图像分类器（CNN卷积神经网络的图片识别篇）...的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。