如何批量预测Faster RCNN模型?

问题描述

我有一个经过训练的 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数组)

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...