查找预测实际测试图像所需的时间

Question

我正在Keras中使用LSTM模型进行图像分类。在预测测试图像时，我想记录预测每个测试图像所需的时间（以毫秒为单位）。有什么方法可以做到？

Answer 1

要计算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秒。这可以看作是粗略估计