yolov7计算fps代码
时间: 2023-11-29 21:44:42 浏览: 110
FPS_YOLOv5功能完整代码
5星 · 资源好评率100%
很抱歉,根据提供的引用内容,没有找到yolov7的计算fps代码。但是,你可以参考以下代码来计算yolov5的fps:
```python
import time
import torch
from models.experimental import attempt_load
from utils.general import non_max_suppression
# 加载模型
model = attempt_load('yolov5s.pt', map_location=torch.device('cpu'))
# 准备输入数据
img = torch.zeros((1, 3, 640, 640), device='cpu') # 假设输入图片大小为640x640
# 预热模型
for i in range(3):
model(img)
# 开始计时
t1 = time.time()
for i in range(100):
# 模型推理
pred = model(img)
# 非极大值抑制
pred = non_max_suppression(pred, conf_thres=0.25, iou_thres=0.45)
# 结束计时
t2 = time.time()
# 计算fps
fps = 100 / (t2 - t1)
print('FPS:', fps)
```
阅读全文