第一步:去PaddleDetection\configs\runtime.yaml将print_flops设置true
在paddle环境安装paddleslim,我是paddlepaddle-gpu==2.3.2但是我安装paddleslim==2.4.0会报错
ImportError: cannot import name '_legacy_C_ops' from 'paddle'
然后我换成2.3.4版本就好了pip install paddleslim==2.3.4
最后python tools/infer.py执行命令就可以看到flops了,最后说明一下我的环境
paddledet==2.5.0
paddlepaddle-gpu==2.3.2
paddleslim==2.3.4