yolov5计算参数量的代码在哪里

yolov5的参数量计算代码在`models/yolo.py`文件中的`ModelEMA`类中的`flops`方法中实现，具体实现如下：

def flops(self, img_size=640):
    from thop import profile
    model = self.model
    x = torch.randn(1, 3, img_size, img_size).cuda()
    model = model.cuda()
    macs, _ = profile(model, inputs=(x,))
    return macs

该方法使用了PyTorch的`thop`库来计算模型的浮点计算次数（FLOPs），并返回该值。其中，`img_size`参数是输入图像的大小，默认为640。方法内部，首先创建了一个输入张量`x`，然后将模型移动到GPU上，最后调用`thop`库的`profile`方法计算模型的FLOPs。