YOLOv7解决无法打印参数量和计算量问题

今天，不晓得为啥子改了代码之后，运行结果不能显示计算量GFLOPS。

解决方案

使用models/yolo.py来打印参数量和计算量。

⭐在models/yolo.py中按下快捷键Ctrl+F搜索model.train()定位过去，然后添加如下代码即可：

    # -------------------打印参数量|计算量---------------- #    from thop import profile    img = torch.rand(1, 3, 640, 640).to(device)    flops, params = profile(model, (img,))    print(f'params: {params}, GFLOPs: {2 * flops}')   # 注意这里的flops要×2，才跟yolo打印出来的值对应得上    print('params: %.2f M, GFLOPs: %.1f B' % (params / 1e6, 2 * flops / 1e9))   # 注意这里的flops要×2，才跟yolo打印出来的值对应得上    # -------------------打印参数量|计算量---------------- #

🔆对于参数量，换算成M的含义是million百万，1M=106

🔆对于每秒浮点运算次数FLOPs，GFLOPs=109FLOPs

最后：

如果你想要进一步了解更多的相关知识，可以关注下面公众号联系~会不定期发布相关设计内容包括但不限于如下内容:信号处理、通信仿真、算法设计、matlab appdesigner，gui设计、simulink仿真......希望能帮到你！