به نام خدا

استفاده از مدل آموزش داده شده

In [1]:
#Load trained model
from keras.models import load_model
model = load_model('cats_and_dogs_small_2.h5')
Using TensorFlow backend.
In [2]:
from keras.preprocessing import image
from matplotlib.pyplot import imshow
import numpy as np
%matplotlib inline
In [11]:
#img_path = 'D:/dataset/catDog/catVsdog/test/cats/cat.1592.jpg'
img_path = 'D:/dataset/catDog/catVsdog/test/dogs/dog.1521.jpg'

img = image.load_img(img_path, target_size=(150, 150))
imshow(img)
x = image.img_to_array(img)
x /= 255.
x = np.expand_dims(x, axis=0)
print(model.predict(x))
print(model.predict_classes(x))
[[0.8157525]]
[[1]]
دوره مقدماتی یادگیری عمیق
علیرضا اخوان پور
پنج شنبه، ۲۵ بهمن ۱۳۹۷
Class.Vision - AkhavanPour.ir - GitHub