将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的输入。

</

最近更新

  1. TCP协议是安全的吗?

    2024-06-11 02:12:03       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-06-11 02:12:03       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-06-11 02:12:03       19 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-06-11 02:12:03       20 阅读

热门阅读

  1. svnadmin备份和还原

    2024-06-11 02:12:03       10 阅读
  2. Perl 运算符

    2024-06-11 02:12:03       11 阅读
  3. 算法:读取redis中指令查询的键

    2024-06-11 02:12:03       8 阅读
  4. 题解web

    题解web

    2024-06-11 02:12:03      9 阅读
  5. 在 React 应用中,怎么封装一个路由权限

    2024-06-11 02:12:03       9 阅读
  6. PHP小方法

    2024-06-11 02:12:03       12 阅读
  7. 课时151:项目发布_基础知识_技术要点

    2024-06-11 02:12:03       10 阅读
  8. C++设计模式---策略模式

    2024-06-11 02:12:03       11 阅读
  9. Permissions 0644 for ‘/home/jsy/.ssh/id_rsa‘ are too open

    2024-06-11 02:12:03       12 阅读
  10. Qt Graphics View Framework 简介

    2024-06-11 02:12:03       10 阅读
  11. CentOS 7 安装配置基础DNS服务,主从域名服务器

    2024-06-11 02:12:03       13 阅读
  12. 【OS】AUTOSAR OS调度器实现原理

    2024-06-11 02:12:03       11 阅读
  13. 智能合约中外部调用漏洞

    2024-06-11 02:12:03       9 阅读
  14. ovs网络配置命令

    2024-06-11 02:12:03       8 阅读
  15. TypeScript基础教程学习

    2024-06-11 02:12:03       11 阅读
  16. ⑤单细胞学习-cellchat组间通讯差异分析

    2024-06-11 02:12:03       11 阅读