我在Keras中使用LSTM模型进行图像分类。在预测测试图像时,我想记录预测每个测试图像所需的时间(以毫秒为单位)。有什么方法可以做到这一点?
要计算Python中的时差,我们使用时间模块。
import time
t1 = time.time()
model.predict( ... )
# Testing has finished
t2 = time.time()
print( 'Time taken was {} seconds'.format( t2 - t1))
对于Jupyter笔记本,请将此标记放在单元格的开头。
%%time
如果假设10个图像的预测需要3秒或3000毫秒,那么对于1个图像,时间将是300毫秒或0.3秒。这可以看作粗略估计