OpenPSG 项目使用教程

OpenPSG 项目使用教程

OpenPSG Benchmarking Panoptic Scene Graph Generation (PSG), ECCV'22 项目地址: https://gitcode.com/gh_mirrors/op/OpenPSG

1. 项目目录结构及介绍

OpenPSG 项目的目录结构如下：

OpenPSG/
├── assets/
├── configs/
├── openpsg/
├── requirements/
├── scripts/
├── tools/
├── .gitignore
├── isort.cfg
├── pre-commit-config.yaml
├── CODE_OF_CONDUCT.md
├── CONTRIBUTING.md
├── LICENSE
├── README.md
├── cog.yaml
├── environment.yml
├── predict.py
├── psg_challenge.md
├── requirements.txt
└── setup.py

目录介绍

• assets/: 存放项目相关的静态资源文件。
• configs/: 存放项目的配置文件。
• openpsg/: 项目的主要代码库，包含核心功能的实现。
• requirements/: 存放项目的依赖文件。
• scripts/: 存放项目的脚本文件，用于执行各种任务。
• tools/: 存放项目的工具文件，提供额外的功能支持。
• .gitignore: Git 忽略文件，指定不需要版本控制的文件和目录。
• isort.cfg: isort 配置文件，用于代码格式化。
• pre-commit-config.yaml: pre-commit 配置文件，用于代码提交前的检查。
• CODE_OF_CONDUCT.md: 项目的行为准则。
• CONTRIBUTING.md: 项目的贡献指南。
• LICENSE: 项目的开源许可证。
• README.md: 项目的介绍文档。
• cog.yaml: cog 配置文件，用于代码生成。
• environment.yml: conda 环境配置文件，用于创建项目的运行环境。
• predict.py: 项目的预测脚本。
• psg_challenge.md: PSG 挑战赛的介绍文档。
• requirements.txt: 项目的依赖文件，列出所有需要的 Python 包。
• setup.py: 项目的安装脚本。

2. 项目启动文件介绍

项目的启动文件主要是 predict.py，该文件用于执行预测任务。启动文件的使用方法如下：

python predict.py

该脚本会根据配置文件中的设置，加载模型并执行预测任务。

3. 项目的配置文件介绍

项目的配置文件主要存放在 configs/ 目录下。配置文件的格式通常为 .yaml 或 .json。以下是一些常见的配置文件及其作用：

• config.yaml: 项目的全局配置文件，包含模型的参数、数据路径等。
• environment.yml: conda 环境配置文件，用于创建项目的运行环境。
• cog.yaml: cog 配置文件，用于代码生成。

配置文件示例

# config.yaml
model:
  type: "PSGModel"
  parameters:
    learning_rate: 0.001
    batch_size: 32

data:
  train_path: "/data/psg/psg_train_val.json"
  val_path: "/data/psg/psg_val_test.json"

通过修改配置文件中的参数，可以调整模型的行为和数据路径。

OpenPSG Benchmarking Panoptic Scene Graph Generation (PSG), ECCV'22 项目地址: https://gitcode.com/gh_mirrors/op/OpenPSG