1,327
社区成员
发帖
与我相关
我的任务
分享
做了个卷积网络判断图片是猫还是狗,最后一层的激活函数是sigmoid,在训练后调用predict函数输出的应该是0-1之间的小数,但结果输出的不是0.0就是1.0。调用predict代码如下:
test_cat_file_path = r"K:\datasets\cat_and_dog\cat3.jpg"
test_image = image.load_img(test_cat_file_path,target_size=(150,150))
test_image = image.img_to_array(test_image)
img = tf.expand_dims(test_image,axis=0)
classes = model.predict(img)
请问问题处在哪里?
请问解决了吗