属性错误:加载TensorFlow H5模型时,“str”对象没有属性“decode”

oknwwptz  于 2023-05-01  发布在  其他
关注(0)|答案(1)|浏览(129)
import numpy as np
import cv2
from tensorflow.keras.models import load_model
model = load_model("my_model_new.h5")

while True:
    success, imgOriginal = cap.read()
    img = np.asarray(imgOriginal)
    img = cv2.resize(img, (32, 32))
    img = preprocessing(img)
    cv2.imshow("Processed Image", img)
    img = img.reshape(1, 32, 32, 1)

    predictions = model.predict(img)
    classIndex = np.argmax(predictions)

    predictions = model.predict(img)
    probVal= np.amax(predictions)
    if probVal > threshold:
        cv2.putText(imgOriginal, str(getClassName(classIndex))+" "+ str(probVal),(50,50),font,1,(0,0,255),1,cv2.LINE_AA)
        cv2.imshow("Original Image", imgOriginal)

    if cv2.waitKey(1) and 0xFF == ord('q'):
        break

这是一个交通标志识别测试部分的部分代码.当我运行代码时,我得到了这个属性的错误消息,我不知道如何解决这个问题。所以有人请告诉我这个错误的修复。

ldfqzlk8

ldfqzlk81#

这个问题是由于Tensorflow和h5 py兼容性中的一个bug。Tensorflow Github issue
您可以安装h5 py版本小于3的tensorflow。0.0(参见注解,但并不总是有效-参见注解)

pip install tensorflow h5py<3.0.0

或者将tensorflow升级到2.4.0以上的版本。

相关问题