1. 安装 thop
1.1 常规安装
pip install thop
1.2 若上述安装方式错误,可以参考以下方式:
pip install thop-i http://pypi.douban.com/simple --trusted-host pypi.douban.com
参考:https://editor.csdn.net/md/?articleId=1064660472. 使用thop计算参数量和计算量
from nets.yolo4 import YoloBody ## 导入模型
from thop import profile ## 导入thop模块
model = YoloBody(3,20)
input = torch.randn(1, 3, 416, 416)
flops, params = profile(model, inputs=(input,))
print('flops',flops) ## 打印计算量
print('params',params) ## 打印参数量
原文链接:https://blog.csdn.net/weixin_44239195/article/details/110390705
统计张量元素个数: