找到预测eact测试图像所需的时间

问题描述 投票:0回答:1

我在Keras中使用LSTM模型进行图像分类。在预测测试图像时,我想记录预测每个测试图像所需的时间(以毫秒为单位)。有什么方法可以做到这一点?

keras
1个回答
0
投票

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

© www.soinside.com 2019 - 2024. All rights reserved.