yolov5打印GFLOPs

为了打印出 YOLOv5 模型的 GFLOPs，你可以使用以下代码：

import torch
from thop import profile

# 导入YOLOv5模型
from models.yolo import Model

# 创建一个模型实例
model = Model()

# 定义输入
input_size = (1, 3, 416, 416)  # （batch_size, channels, height, width）
input_data = torch.randn(*input_size)

# 使用profile函数计算GFLOPs和参数数量
flops, params = profile(model, inputs=(input_data,), verbose=False)

# 打印GFLOPs
print(f"GFLOPs: {flops / 1e9}")

请确保将 `models.yolo` 替换为正确的 YOLOv5 模型路径。运行代码后，你将打印出 YOLOv5 模型的 GFLOPs。