代码如下:
from imageai.Prediction import ImagePrediction
import os
execution_path = os.getcwd()
prediction = ImagePrediction()
prediction.setModelTypeAsResNet()
prediction.setModelPath(os.path.join(execution_path,"h5model/resnet50_weights_tf_dim_ordering_tf_kernels.h5"))
prediction.loadModel()
predictions, probabilities = prediction.predictImage(os.path.join(execution_path,"1.jpg"))
for eachPrediction, eachProbability in zip(predictions,probabilities):
print(eachPrediction, ":", eachProbability)
报错内容如下
环境:
tensorflow-gpu 2.0.0
scipy 1.3.1
keras 2.1.5
在stackoverflow找到了类似错误,作者说是没有进行图片黑白化,但是我黑白化之后还是会报这样的错误,求教各位大神。