grasp-anything
收藏数据集概述
数据集名称
- Grasp-Anything
数据集访问
- 访问链接:Grasp-Anything数据集
数据集用途
- 用于训练和测试语言驱动的抓取检测模型。
数据集训练命令
-
使用GR-ConvNet训练: bash $ python -m torch.distributed.launch --nproc_per_node=<num_gpus> --use_env -m train_network_diffusion --dataset grasp-anywhere --dataset-path data/grasp-anything++/ --add-file-path data/grasp-anything++/seen --description training_grasp_anything++_lgd --use-depth 0 --seen 1 --network lgd --epochs 1000
-
训练其他网络的语言版本: bash $ python train_network.py --dataset grasp-anywhere --dataset-path data/grasp-anything/ --add-file-path data/grasp-anything++/seen --description <description> --use-depth 0 --seen 1 --network <network_name>
-
训练其他基准模型: bash $ python train_network.py --dataset <dataset> --dataset-path <dataset> --description <your_description> --use-depth 0 --network <baseline_name>
数据集测试命令
-
测试不同基准模型: bash $ python -m torch.distributed.launch --nproc_per_node=1 --use_env -m evaluate_diffusion --dataset grasp-anywhere --dataset-path data/grasp-anything++/ --add-file-path data/grasp-anything++/seen --iou-eval --seen 1 --use-depth 0 --network <path_to_pretrained_network>
-
或 bash $ python evaluate.py --dataset grasp-anywhere --dataset-path data/grasp-anything --add-file-path data/grasp-anything++/seen --iou-eval --seen 0 --use-depth 0 --network <path_to_pretrained_network>
预训练模型
- 预训练模型权重下载链接:预训练模型




