Iya*_*man 6 python-3.x tensorflow
我正在关注本教程系列:https://www.youtube.com/watch? v=A4K6D_gx2Iw&list=PLQVvvaa0QuDfhTox0AjmQ6tvTgMBZBEXN&index=6
当我尝试通过预测在程序外部使用模型时,出现以下错误:错误:OpenCV(4.1.0) /Users/travis/build/skvark/opencv-python/opencv/modules/imgproc/src/resize。 cpp:3718:错误:(-215:断言失败)!函数“调整大小”中的 ssize.empty()
加载模型时的代码如下:
import cv2
import tensorflow as tf
CATEGORIES = ["Dog", "Cat"]
import cv2
import tensorflow as tf
CATEGORIES = ["Dog", "Cat"]
def prepare(filepath):
IMG_SIZE = 50
img_array = cv2.imread(filepath, cv2.IMREAD_GRAYSCALE)
#return img_array.reshape(-1, IMG_SIZE, IMG_SIZE, 1)
new_array = cv2.resize(img_array, (IMG_SIZE, IMG_SIZE))
return new_array.reshape(-1, IMG_SIZE, IMG_SIZE, 1)
#
model = tf.keras.models.load_model('iyad')
#
predication = model.predict([prepare("Dog.jpg")])
print(predication)
Run Code Online (Sandbox Code Playgroud)
小智 4
它告诉你原始图像的大小为0,对应于你从 读取时的大小img_array = cv2.imread(filepath, cv2.IMREAD_GRAYSCALE)。检查您是否已Dog.jpg加载该图片。