Grasp-Anything 项目使用教程

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

Grasp-Anything/
├── inference/
│   └── models/
│       └── grasp_det_seg/
├── script/
├── split/
├── utils/
├── weights/
├── .gitignore
├── LICENSE
├── README.md
├── _config.yml
├── cleanup.sh
├── evaluate.py
├── requirements.txt
├── train_network.py
└── train_network_grasp_det_seg.py

目录结构介绍

• inference/: 包含推理相关的模型文件，特别是 grasp_det_seg 模型。
• script/: 包含项目的脚本文件。
• split/: 包含数据集分割相关的文件。
• utils/: 包含项目中使用的各种实用工具。
• weights/: 包含预训练的权重文件。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目的开源许可证文件。
• README.md: 项目的介绍和使用说明。
• _config.yml: 项目的配置文件。
• cleanup.sh: 清理脚本。
• evaluate.py: 评估脚本。
• requirements.txt: 项目依赖的 Python 包列表。
• train_network.py: 训练网络的主脚本。
• train_network_grasp_det_seg.py: 训练 grasp detection 和 segmentation 网络的脚本。

2. 项目的启动文件介绍

train_network.py

这是项目的主要启动文件，用于训练网络。可以通过以下命令启动训练：

python train_network.py --dataset <dataset> --dataset-path <dataset> --description <your_description> --use-depth 0

evaluate.py

用于评估训练好的模型。可以通过以下命令启动评估：

python evaluate.py --network <path_to_pretrained_network> --dataset <dataset> --dataset-path data/<dataset> --iou-eval

3. 项目的配置文件介绍

_config.yml

这是项目的配置文件，包含了项目的各种配置选项。以下是配置文件的部分内容示例：

# 示例配置文件内容
dataset: cornell
dataset-path: data/cornell
description: training_cornell
use-depth: 0
network: ggcnn

配置文件说明

• dataset: 指定要使用的数据集名称。
• dataset-path: 数据集的路径。
• description: 训练描述，用于记录训练信息。
• use-depth: 是否使用深度信息，0 表示不使用，1 表示使用。
• network: 指定要使用的网络模型，如 ggcnn。

通过这些配置文件，可以灵活地调整项目的运行参数，以适应不同的训练和评估需求。