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

将AIRNet集成到yolov8中,实现端到端训练与推理

AIRNet是一个图像修复网络,支持对图像进行去雾、去雨、去噪声的修复。其基于对比的退化编码器(CBDE),将各种退化类型统一到同一嵌入空间;然后,基于退化引导恢复网络(DGRN)将嵌入空间修复为目标图像。可以将AIRNet的输出与yolov8进行端到端集成,实现部署上的简化。 本博文的实现,支持AIRNet模型在yolov8中的推理与训练,但是对于AIRNet模型的loss,不支持在训练输出中显示(改动量较大,故放弃了),但在每一次iter迭代中,各位可以添加loss信息存储代码,在训练结束后自行将item对应的loss计算为epoch的loss。

补充说明:320x320的输入,batch为2,12g显存不支持AIRNet-yolov8的训练。

要集成到yolov9中的话,参考https://hpg123.blog.csdn.net/article/details/137935493中的配置文件即可。

在这里插入图片描述
这里的实现,是以yolov8的训练数据格式为准,要想支持去雨、去雾、去噪声的训练,则需要修改输入的im_tensor,自行对im_tensor添加雨、雾、噪声等。
将AIRNet与yolov9进行集成后,发现显存占用比较多,故而在推理过程中使用320x320的输入。

</

http://www.lryc.cn/news/368742.html

相关文章:

  • hcache缓存查看工具
  • Java 数据类型 -- Java 语言的 8 种基本数据类型、字符串与数组
  • kafka-生产者事务-数据传递语义事务介绍事务消息发送(SpringBoot整合Kafka)
  • 免费!GPT-4o发布,实时语音视频丝滑交互
  • DevOps的原理及应用详解(四)
  • 关于选择,关于处事
  • 大话设计模式解读02-策略模式
  • 展会邀请 | 龙智即将亮相2024上海国际嵌入式展,带来安全合规、单一可信数据源、可追溯、高效协同的嵌入式开发解决方案
  • codeforce round951 div2
  • arcgis开发记录
  • RPA-UiBot6.0数据整理机器人—杂乱数据秒变报表
  • Application UI
  • 关于 Redis 中集群
  • C++必修:探索C++的内存管理
  • python列表---基本语法(浅拷贝,深拷贝等)
  • go语言接口之sort.Interface接口
  • android:text 总为大写字母的原因
  • CISCN2024 初赛 wp 部分复现(Re)
  • YOLOv10、YOLOv9 和 YOLOv8 在实际视频中的对比
  • 热题系列章节5
  • ArcGIS for js 4.x 加载图层
  • Three.js和Babylon.js,webGL中的对比效果分析!
  • flask实现抽奖程序(一)
  • Python中数据库连接的管理
  • 【JAVA技术】mybatis 数据库敏感字段加解密方案
  • Collections工具类及其案例
  • Duck Bro的第512天创作纪念日
  • 【机器学习】GPT-4中的机器学习如何塑造人类与AI的新对话
  • 晨控CK-UR12-E01与欧姆龙NX/NJ系列EtherNet/IP通讯手册
  • 模板显式、隐式实例化和(偏)特化、具体化的详细分析