神经网络已经为您训练完成!

使用代码如下:

import keras
from keras.models import load_model  
model = load_model('简单的猫狗cnn.h5')
model.summary()

如何进行识别?

from matplotlib import pyplot as plt
import numpy as np
path = 'testimg/QQ图片20210802200354.jpg' #图片路径
img_height, img_width = 150,150 #大小改为固定格式
img_keras = keras.preprocessing.image.load_img(path=path, target_size=(img_height, img_width))
plt.imshow(img_keras)
img_keras =  np.array(img_keras)
img_keras = img_keras/255
img_keras = np.expand_dims(img_keras, axis=0)
img_keras.reshape(150,150,3)
print('狗狗的相似度是',model.predict(img_keras)[0][0])
print('猫猫的相似度是',1-model.predict(img_keras)[0][0])
Last modification:August 3, 2021
如果觉得我的内容对你有用,请随意赞赏