pytorch 45 使用自己的数据对groundingdino进行微调

本文主要是介绍pytorch 45 使用自己的数据对groundingdino进行微调，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

基于项目：https://github.com/Asad-Ismail/Grounding-Dino-FineTuning 完成finetuning操作

Grounding DINO是一种基于transformer的视觉语言检测方法，它可以根据文字描述检测指定目标。它是在DINO的基础上，增加了多个阶段的视觉语言模态融合，包括特征增强器、语言指导的查询选择模块和跨模态解码器。它可以实现最先进的对象检测器的性能，消除了对NMS等手工模块的依赖。然而在应用到业务数据上还是有所不足，故而需要对Grounding DINO进行微调。

本博文基于Grounding-Dino-FineTuning项目实现对自己yolo数据的微调，在使用GroundingDino需要下载外网资源，在训练数据时目前仅支持batch为1的训练，导致显存利用率不高，后续会考虑优化Grounding-Dino-FineTuning项目。

1、环境准备

1.1 预训练代码下载

下载 https://github.com/Asad-Ismail/Grounding-Dino-FineTuning 项目，并解压
下载 https://github.com/IDEA-Research/GroundingDINO项目，并解压。

然后使用GroundingDINO项目的groundingdino目录复制到Grounding-Dino-FineTuning根目录下&#

这篇关于pytorch 45 使用自己的数据对groundingdino进行微调的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---