当前位置: 首页 > news >正文

YOLOv9模型训练或测试过程中,无法打印模型的GFLOPs

项目场景:

YOLOv9模型的改进中,常常需要替换一些模块来提高模型的精度。但在评价模型大小规模的时候需要根据模型的参数量、计算量进行评定,一般在模型的训练文件train.py,或者是test.py还有models/yolo.py都会输出这些数据。


问题描述

在模型改进过程中,在YOLOv9的基础模型中添加了新模块,在训练时,打印店信息中没有打印参数量GFLOPs。在YOLOv7的改进中,未打印GFLOPs的只需要安装thop包就可以正常打印了。但在YOLOv9中似乎不行。例如此处运行models/yolo.py,未打印GFLOPs

在这里插入图片描述


解决方案:

models/yolo.pyif __name__ == '__main__'下方添加如下代码,自行输出:

from thop import profile
img = torch.rand(1, 3, 640, 640).to(device)
flops, params = profile(model, (img,)) 
print('parameters: %.2f M, GFLOPs: %.1f GFLOPs' % (params / 1e6, 2 * flops / 1e9))  

具体添加位置:

在这里插入图片描述

运行结果:

在这里插入图片描述
此时就已经输出GFLOPs了,并且输出的参数量也是一致的。

相关文章:

  • 北京网站建设多少钱?
  • 辽宁网页制作哪家好_网站建设
  • 高端品牌网站建设_汉中网站制作
  • shell脚本编程(数组)
  • 智能家居系统(基于STM32F103C8T6标准库+FreeRTOS+Qt串口开发实现)
  • 【系统架构设计师-2010年】综合知识-答案及详解
  • Java 7.3 - 分布式 id
  • QT教程:start()和startTimer()的区别
  • 第五章 Pytorch完成线性回归
  • 小白入门LLM大模型最牛X教程------上交《动手学大模型应用开发》!
  • 语音测试(一)ffmpeg视频转音频
  • 2024年全国大学生数学建模竞赛(A题) 建模解析|“板凳龙” 闹元宵|小鹿学长带队指引全代码文章与思路
  • OpenHarmony轻松玩转GIF数据渲染
  • 语言的自动类型
  • QueryWrapper中的or和and
  • 磁盘加密工具 | VeraCrypt v1.26.15 绿色版
  • STM32基础篇:RTC × Unix时间戳 × BKP
  • [Deepin] 简单使用 RustDesk 实现远程访问Deepin
  • 网络传输文件的问题
  • 5分钟即可掌握的前端高效利器:JavaScript 策略模式
  • ES6之路之模块详解
  • Flannel解读
  • github从入门到放弃(1)
  • iOS筛选菜单、分段选择器、导航栏、悬浮窗、转场动画、启动视频等源码
  • js ES6 求数组的交集,并集,还有差集
  • MySQL数据库运维之数据恢复
  • Node.js 新计划:使用 V8 snapshot 将启动速度提升 8 倍
  • PHP 的 SAPI 是个什么东西
  • Python学习笔记 字符串拼接
  • vue从入门到进阶:计算属性computed与侦听器watch(三)
  • 从伪并行的 Python 多线程说起
  • 基于Dubbo+ZooKeeper的分布式服务的实现
  • 快速构建spring-cloud+sleuth+rabbit+ zipkin+es+kibana+grafana日志跟踪平台
  • 深入体验bash on windows,在windows上搭建原生的linux开发环境,酷!
  • 问题之ssh中Host key verification failed的解决
  • 3月7日云栖精选夜读 | RSA 2019安全大会:企业资产管理成行业新风向标,云上安全占绝对优势 ...
  • 不要一棍子打翻所有黑盒模型,其实可以让它们发挥作用 ...
  • 容器镜像
  • ​14:00面试,14:06就出来了,问的问题有点变态。。。
  • ​软考-高级-系统架构设计师教程(清华第2版)【第20章 系统架构设计师论文写作要点(P717~728)-思维导图】​
  • ‌前端列表展示1000条大量数据时,后端通常需要进行一定的处理。‌
  • #[Composer学习笔记]Part1:安装composer并通过composer创建一个项目
  • (1)Jupyter Notebook 下载及安装
  • (12)Linux 常见的三种进程状态
  • (C11) 泛型表达式
  • (Matalb时序预测)PSO-BP粒子群算法优化BP神经网络的多维时序回归预测
  • (八)Spring源码解析:Spring MVC
  • (算法)Game
  • (转)JVM内存分配 -Xms128m -Xmx512m -XX:PermSize=128m -XX:MaxPermSize=512m
  • (转)VC++中ondraw在什么时候调用的
  • (转)菜鸟学数据库(三)——存储过程
  • . Flume面试题
  • .net 7 上传文件踩坑
  • .Net Winform开发笔记(一)
  • .NET 编写一个可以异步等待循环中任何一个部分的 Awaiter
  • .NET 将混合了多个不同平台(Windows Mac Linux)的文件 目录的路径格式化成同一个平台下的路径
  • .NET国产化改造探索(三)、银河麒麟安装.NET 8环境
  • .net专家(张羿专栏)