if torch.cuda.is_availabel():
cuda = True
...
fps = 0
i = 0
if cuda = True:
torch.cuda.synchronize()
start_time = time.time()
...
...
if cuda = True:
torch.cuda.synchronize()
secend_time = time.time()
use_time = second_time - start_time
if i == 0:
fps = 1/use_time
fps = (fps + 1/use_time) / 2
i += 1
