Calculation model FPS

Whether it is image classification or , object detection / Instance segmentation , Detection speed is an important indicator .

Calculation model FPS The process ：
Just add one before and after the model runs time Just record the time , Then calculate the model through the time difference FPS

start = time.time()
with torch.no_grad():
	seg_result = model.forward(img, target)
img_show = show_result_ins(imgpath, seg_result)
end = time.time()
t_all = end - start
# print("spend time: ", t_all ,"s")
print('average time:{:.02f} s'.format(np.mean(t_all) / 1))
print('average FPS :{:.02f} fps'.format(1 / np.mean(t_all)))

The complete code can be referred to as follows

import time
import torch
import numpy as np

net = build_model
net.eval()

# x Is the size of the input picture 
x = torch.zeros((1,3,H,W)).cuda()
t_all = []

for i in range(100):
    t1 = time.time()
    y = net(x)
    t2 = time.time()
    t_all.append(t2 - t1)

print('average time:', np.mean(t_all) / 1)
print('average fps:',1 / np.mean(t_all))

print('fastest time:', min(t_all) / 1)
print('fastest fps:',1 / min(t_all))

print('slowest time:', max(t_all) / 1)
print('slowest fps:',1 / max(t_all))