参数量与图片尺寸无关
而计算复杂度(GFlops)与输入图片的尺寸有关,作比较的时候要确保,输入尺寸一致的,最新版本的mmdetection的tools/analysis_tools/get_flops.py中不支持更改输入图片尺寸,而是自己从数据集中获取大小,想要自定义输入尺寸可以按如下操作:
①找到.conda/envs/mmlab/lib/python3.8/site-packages/mmengine/analysis/print_helper.py文件
②在677行get_model_complexity_info函数的最开始加上以下两句
input_shape=(3,640,640)
inputs=None
本文介绍了在MMDetection的最新版本中,如何在get_flops.py工具中自定义输入图片尺寸以进行计算复杂度分析。提到需修改mmengine/analysis/print_helper.py文件中的input_shape和inputs变量以确保在比较时输入尺寸一致。
2783

被折叠的 条评论
为什么被折叠?



