本文主要是介绍修改yolov9的模型打印不出来Gflops的解决办法,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
正在修改yolov9的模块,发现修改后的模型没有GFlops这个参数
解决办法:
找到utils/torch_utils.py这个文件,有一个model_info函数
然后将其中的stride改为固定的640就可以打印了。
stride = max(int(model.stride.max()), 32) if hasattr(model, 'stride') else 32 # max stride
改成:
stride = 640
结果如下:
这篇关于修改yolov9的模型打印不出来Gflops的解决办法的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!