GroundingDINO 项目使用教程

GroundingDINO 项目使用教程

GroundingDINO论文 'Grounding DINO: 将DINO与基于地面的预训练结合用于开放式目标检测' 的官方实现。项目地址:https://gitcode.com/gh_mirrors/gr/GroundingDINO

1. 项目的目录结构及介绍

GroundingDINO 项目的目录结构如下：

GroundingDINO/
├── demo/
│   └── test_ap_on_coco.py
├── groundingdino/
│   └── config/
│       └── GroundingDINO_SwinT_OGC.py
├── weights/
│   └── groundingdino_swint_ogc.pth
├── README.md
├── LICENSE
└── requirements.txt

目录结构介绍

• demo/: 包含项目的演示脚本，如 test_ap_on_coco.py。
• groundingdino/: 核心代码目录，包含配置文件和其他相关代码。
  • config/: 配置文件目录，如 GroundingDINO_SwinT_OGC.py。
• weights/: 模型权重文件目录，如 groundingdino_swint_ogc.pth。
• README.md: 项目说明文档。
• LICENSE: 项目许可证文件。
• requirements.txt: 项目依赖文件。

2. 项目的启动文件介绍

项目的启动文件主要是 demo/test_ap_on_coco.py。该文件用于在 COCO 数据集上测试模型的平均精度（AP）。

启动文件介绍

• test_ap_on_coco.py:
  • 功能：在 COCO 数据集上进行模型测试。
  • 使用方法：通过命令行运行该脚本，并指定相关参数，如配置文件路径、权重文件路径、注释文件路径和图像目录路径。

示例命令：

python demo/test_ap_on_coco.py \
  -c groundingdino/config/GroundingDINO_SwinT_OGC.py \
  -p weights/groundingdino_swint_ogc.pth \
  --anno_path /path/to/annotations/instances_val2017.json \
  --image_dir /path/to/imagedir/val2017

3. 项目的配置文件介绍

项目的配置文件位于 groundingdino/config/ 目录下，主要配置文件是 GroundingDINO_SwinT_OGC.py。

配置文件介绍

• GroundingDINO_SwinT_OGC.py:
  • 功能：定义模型的配置参数，包括模型结构、训练参数、数据集路径等。
  • 主要参数：
    • backbone: 模型骨干网络，如 Swin-T。
    • data: 数据集相关配置，如 COCO 数据集路径。
    • training: 训练相关配置，如学习率、批大小等。

配置文件的使用方法：

1. 在启动脚本中指定配置文件路径。
2. 根据需要修改配置文件中的参数。

示例配置文件路径指定：

python demo/test_ap_on_coco.py \
  -c groundingdino/config/GroundingDINO_SwinT_OGC.py \
  ...

通过以上介绍，您可以了解 GroundingDINO 项目的目录结构、启动文件和配置文件的基本使用方法。希望这份教程对您有所帮助。

GroundingDINO论文 'Grounding DINO: 将DINO与基于地面的预训练结合用于开放式目标检测' 的官方实现。项目地址:https://gitcode.com/gh_mirrors/gr/GroundingDINO