mmsegmentation 计算参数量（params）和计算量（ FLOPs）

这命令输入了好几次都错了，容易忘，记录一下用法

官方命令

python tools/get_flops.py ${CONFIG_FILE} [--shape ${INPUT_SHAPE}]
得到结果：

==============================
Input shape: (3, 2048, 1024)
Flops: 1429.68 GMac
Params: 48.98 M
==============================

来源：mmsegmentation-常用工具

简单分析

1. shape后跟着int类型，下面表示输入一个图片大小数字就会返回一个（3,shape,shape）的元组
2. 如果不输入–shape及参数，就默认按照输入图片是2048*1024大小展开计算

实际操作

使用的图片是512*512，故输入512即可
python tools/get_flops.py xxx/config_path --shape 512

==============================
Input shape: (3, 512, 512)
Flops: 253.9 GFLOPs
Params: 64.29 M
==============================

mark一下，后面再细看怎么计算这些参数的