Pytorch计算模型的参数量和计算量

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=106466047

2. 使用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)			## 打印参数量

附一张自己模型计算得到的结果：

参考链接：
[Pytorch]如何获取模型的参数量和计算量
在Pytorch中计算自己模型的FLOPs方式