Grasp-Anything
收藏Grasp-Anything 数据集概述
数据集访问
- 数据集可通过以下链接获取:Grasp-Anything 数据集下载
数据集使用
训练
-
默认使用 GR-ConvNet 网络进行训练。训练命令示例: bash $ python train_network.py --dataset <dataset> --dataset-path <dataset> --description <your_description> --use-depth 0
例如,训练 GR-ConvNet 在 Cornell 数据集上的命令: bash $ python train_network.py --dataset cornell --dataset-path data/cornell --description training_cornell --use-depth 0
-
支持其他基准网络的训练,命令示例: bash $ python train_network.py --dataset <dataset> --dataset-path <dataset> --description <your_description> --use-depth 0 --network <baseline_name>
例如,训练 GG-CNN 在 Cornell 数据集上的命令: bash python train_network.py --dataset cornell --dataset-path data/cornell/ --description training_ggcnn_on_cornell --use-depth 0 --network ggcnn
测试
-
测试命令示例: bash python evaluate.py --network <path_to_pretrained_network> --dataset <dataset> --dataset-path data/<dataset> --iou-eval
其中,
<path_to_pretrained_network>是预训练模型的路径,通常存储在logs/<timstamp>_<training_description>。预训练权重可从以下链接获取:预训练模型下载




