正在修改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这个参数
找到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
结果如下:
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1524010.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!