如何计算花费在CNN每个层上的花费时间和caffe界面？

Question

我试图计算在fast-rcnn网络中每一层花费的时间。我发现咖啡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) 

花费的时间