YOLO改进模块后不出现Glops参数问题

改进模块成功运行但是Glops不显示

在train.py这个文件的最上端添加这个代码即可实现

python 复制代码
import torch
from thop import profile
from models.yolo import Model  # 假设 Model 定义在 yolo.py 中

# 创建模型
 model = Model(cfg='E:/dyh/yolov5/models/odconv.yaml')  # 根据你的配置文件


 inputs = torch.randn(1, 3, 640, 640)
 macs, params = profile(model, inputs=(inputs,))

# 转换为 FLOPs
 flops = macs * 2
 print(f"FLOPs: {flops / 1e9-0.6} GFLOPs")  # 转换为 GFLOPs
#

这里面的-0.6是我根据正常的模块运行后和他之间计算出来的误差,大家可以自己尝试误差是多少然后跟改即可

相关推荐
CV实验室1 小时前
TIP 2025 | 哈工大&哈佛等提出 TripleMixer:攻克雨雪雾干扰的3D点云去噪网络!
人工智能·计算机视觉·3d·论文
余俊晖2 小时前
一套针对金融领域多模态问答的自适应多层级RAG框架-VeritasFi
人工智能·金融·rag
码农阿树2 小时前
视频解析转换耗时—OpenCV优化摸索路
人工智能·opencv·音视频
伏小白白白3 小时前
【论文精度-2】求解车辆路径问题的神经组合优化算法:综合展望(Yubin Xiao,2025)
人工智能·算法·机器学习
应用市场3 小时前
OpenCV编程入门:从零开始的计算机视觉之旅
人工智能·opencv·计算机视觉
星域智链4 小时前
宠物智能用品:当毛孩子遇上 AI,是便利还是过度?
人工智能·科技·学习·宠物
taxunjishu4 小时前
DeviceNet 转 MODBUS TCP罗克韦尔 ControlLogix PLC 与上位机在汽车零部件涂装生产线漆膜厚度精准控制的通讯配置案例
人工智能·区块链·工业物联网·工业自动化·总线协议
说私域4 小时前
基于多模态AI技术的传统行业智能化升级路径研究——以开源AI大模型、AI智能名片与S2B2C商城小程序为例
人工智能·小程序·开源
囚生CY4 小时前
【速写】优化的深度与广度(Adam & Moun)
人工智能·python·算法
hqyjzsb4 小时前
2025年市场岗位能力重构与跨领域转型路径分析
c语言·人工智能·信息可视化·重构·媒体·改行学it·caie