问题描述
我有一个经过训练的 RCNN (Keras-Retinanet) 模型,我一次只能预测一张图片。
Boxes,scores,labels = model.predict_on_batch(np.expand_dims(image,axis=0))
完整脚本为 here。
有没有办法一次预测多个图像?
谢谢
解决方法
扩大暗淡的原因是人为地添加批次维度。只需将一堆图像堆叠在一起并将其传递给此函数即可获得一批结果。
换种说法,现在你正在传入:
[图片]。
您可以改为传入:
[image_1,image_2,image_3,...](作为一个numpy,而不是python数组)