Grounded SAM 2 项目安装与使用指南
1. 项目目录结构及介绍
Grounded SAM 2 项目的目录结构如下:
.
├── assets/ # 存放项目所需的资源文件
├── checkpoints/ # 存放预训练模型和训练结果
├── demo/ # 包含项目演示脚本
├── demo_images/ # 存放演示图片
├── gdino_checkpoints/ # Grounding DINO 模型的预训练权重
├── grounding_dino/ # Grounding DINO 相关代码
├── notebooks/ # Jupyter notebooks 用于分析和可视化
├── sam2/ # SAM 2 相关代码和配置
├── sav_dataset/ # 存放数据集相关文件
├── tools/ # 项目工具脚本
├── training/ # 模型训练相关脚本和代码
├── utils/ # 通用工具和辅助函数
├── .clang-format # Clang 格式配置文件
├── .gitignore # Git 忽略文件
├── .watchmanconfig # Watchman 配置文件
├── CODE_OF_CONDUCT.md # 项目行为准则
├── CONTRIBUTING.md # 贡献指南
├── Dockerfile # Docker 配置文件
├── INSTALL.md # 安装指南
├── LICENSE # 项目许可证
├── LICENSE_cctorch # cctorch 许可证
├── LICENSE_groundingdino # Grounding DINO 许可证
├── LICENSE_sam2 # SAM 2 许可证
├── MANIFEST.in # 包含文件列表
├── Makefile # Makefile 文件
├── README.md # 项目说明文件
├── SAM2_README.md # SAM 2 说明文件
├── backend.Dockerfile # 后端服务 Docker 配置文件
├── docker-compose.yaml # Docker Compose 配置文件
├── grounded_sam2_* # 项目演示和测试脚本
├── pyproject.toml # Python 项目配置文件
└── setup.py # Python 包设置文件
2. 项目的启动文件介绍
项目的启动文件主要是 demo/
目录下的脚本,这些脚本用于演示 Grounded SAM 2 的各项功能。以下是一些主要的启动文件:
grounded_sam2_dinox_demo.py
:使用 DINO-X 进行演示的脚本。grounded_sam2_florence2_autolabel_pipeline.py
:使用 Florence-2 进行自动标注管道的演示脚本。grounded_sam2_hf_model_demo.py
:使用 Hugging Face 模型进行演示的脚本。grounded_sam2_local_demo.py
:使用本地模型进行演示的脚本。grounded_sam2_tracking_demo.py
:演示视频跟踪功能的脚本。
3. 项目的配置文件介绍
项目的配置文件主要用于设置模型参数、路径和其他相关信息。以下是一些主要的配置文件:
config.py
:主配置文件,包含模型、数据集、训练和测试的设置。train_config.yaml
:训练配置文件,定义了训练过程中的参数设置。test_config.yaml
:测试配置文件,定义了测试过程中的参数设置。
用户需要根据自己的需求修改这些配置文件,以确保项目能够正确运行。在修改配置文件时,请参考项目官方文档和注释以了解每个参数的具体含义和作用。