如何用caffe接口计算CNN中每一层花费的时间？

Question

我试图计算fast-rcnn网络中每一层花费的时间。我发现caffe cmd接口caffe time可以做到这一点。但是fast-rcnn是基于python代码的，python接口中是否有任何函数可以帮我计算时间？

Answer 1

python有分析工具。您可以查看timeit和cProfile工具 如果您正在寻找更基本的东西，可以使用time模块来测量时间：

import time
t = time.time()
run_rcnn_script()
dt = time.time()-t
print "time spent in run_rcnn_script: ", dt, " sec."

Answer 2

计算每层花费的时间

import timeit

t1=timeit.default_timer()
net.forward(start='start_layer_name',end='end_layer_name')
t2=timeit.default_timer()

print 'time is {}'.format(t2-t1)