創(chuàng)新互聯(lián)www.cdcxhl.cn八線動態(tài)BGP香港云服務(wù)器提供商,新人活動買多久送多久,劃算不套路!
這篇文章主要介紹在keras中對單一圖像進行預(yù)測并返回預(yù)測結(jié)果的案例,文中介紹的非常詳細,具有一定的參考價值,感興趣的小伙伴們一定要看完!
模型經(jīng)過訓(xùn)練測試之后,我們往往用一兩張圖對模型預(yù)測結(jié)果進行分析討論,那么下面介紹在keras中用已訓(xùn)練的模型經(jīng)過測試的方法。
下面是以利用預(yù)訓(xùn)練的ResNet來展示預(yù)測的效果,選了一張狗的圖片,是來自一個kaggle比賽的。
預(yù)測結(jié)果第一個是一種蘇格蘭品種的狗,我也不知道準不準 == 。
import numpy as np from keras.applications.imagenet_utils import decode_predictions from keras.preprocessing import image from keras.applications import * import os # 忽略硬件加速的警告信息 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' file_path = 'images/0a70f64352edfef4c82c22015f0e3a20.jpg' img = image.load_img(file_path, target_size=(224, 224)) x = image.img_to_array(img) x = np.expand_dims(x, axis=0) model = ResNet50(weights='imagenet') y = model.predict(x) # print(np.argmax(y)) print('Predicted:', decode_predictions(y, top=3)[0])