我正在尝试使图像分类器工作。到目前为止,该模型似乎确实可以正常工作,但是现在每次我要测试图像以查看图像是否被正确识别时,我都必须进行整个训练。我对此很陌生,但我想应该有另一种方法来只对图像进行测试而无需培训,对吗?
关于代码本身,我还有一个问题。
if result [0][0] >= 0.5:
prediction = "cogwheel"
else:
prediction = "not a cogwheel"
print(prediction)
Run Code Online (Sandbox Code Playgroud)
我试图区分代表齿轮的图像和不代表齿轮的图像。我知道,如果概率大于0.5,则是齿轮,否则就不是。但是[0] [0]是什么意思?
非常感谢你的帮助!