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

将任意图像增强模型与ultralytics中任意模型进行结合,支持自定义图像增强模块的loss,实现端到端训练

图像增强模型算法版本较多,同时ultralytics库的迭代也比较大,直接将图像增强模型算法添加到ultralytics库中可能只是当前版本支持,在ultralytics库升级后,又需要重新配置。故而分享一种方式进行实现,支持任意图像增强模型与ultralytics库中任务模型(如yolov8,yolov9,yolov10等模型)进行结合。支持模型的训练、验证与导出。

其核心就是,对模型结构进行替换,同时确保调用ultralytics库接口的train函数时模型结构替换依旧有效。
虽然没有对gdip模块进行测试,但确认这种修改方式是支持实现gdip-yolov10的,也支持实现ia-yolov10。

1、保存图像增强模型

如sci++项目,将模型结构带权重保存为pth文件即可,具体参考示意如下所示


from model import Finetunemodel
import torchif __name__ ==<

http://www.mrgr.cn/news/33783.html

相关文章:

  • Tomcat中间件常见漏洞复现
  • #面试系列-腾讯后端一面
  • 思维商业篇(4)—产业上下游定
  • AJAX入门
  • 2024PDF内容修改秘籍:工具推荐与技巧分享
  • 【C++笔记】C++编译器拷贝优化和内存管理
  • 【VLM小白指北 (1) 】An Introduction to Vision-Language Modeling
  • 【Makefile】linux学习总结
  • 2024年华为杯数学建模E题-高速公路应急车道启用建模-基于YOLO8的数据处理代码参考(无偿分享)
  • 如何选择与高效使用编程工具提升工作效率
  • JavaScript是如何来的~~
  • 《C++中的原子操作:实现高效并发编程的关键》
  • 面向对象程序设计——set容器の简析
  • Python|OpenCV-实现识别目标图像中的圆圈(20)
  • cv中每个patch的关联
  • C++ Linux IPC进程通信-消息队列MQ
  • ZYNQ学习--AXI总线协议
  • CSS 的继承性、层叠性与权重问题解析
  • Python办公自动化教程(003):PDF的加密
  • FreeRTOS通过消息队列+信号量实现串口命令解析(串口中断+空闲中断)