├── .gitignore ├── LICENSE ├── README.md ├── count.py ├── dataset_utils ├── eval_score │ ├── configs │ │ └── config.py │ ├── eval.py │ └── eval_utils │ │ ├── evaluation_data_generator.py │ │ ├── pointcloud.py │ │ └── torch_scene_point_cloud.py ├── get_regiondataset.py └── scoredataset.py ├── markdown ├── 1.png ├── 1_1.png ├── 2.png ├── 3.png ├── 4.png ├── 4_2.png ├── score2.png ├── score3.png ├── vgr2.png └── vgr3.png ├── multi_model ├── gripper_region_network.py ├── score_network.py └── utils │ ├── pn2_utils │ ├── __init__.py │ ├── csrc │ │ ├── ball_query.h │ │ ├── ball_query_kernel.cu │ │ ├── grouping.h │ │ ├── grouping_kernel.cu │ │ ├── interpolate.h │ │ ├── interpolate_kernel.cu │ │ ├── main.cpp │ │ ├── sampling.h │ │ └── sampling_kernel.cu │ ├── function.py │ ├── functions │ │ ├── __init__.py │ │ ├── csrc │ │ │ ├── gather_knn.h │ │ │ ├── gather_knn_kernel.cu │ │ │ └── main.cpp │ │ ├── gather_knn.py │ │ └── setup.py │ ├── modules.py │ ├── nn │ │ ├── __init__.py │ │ ├── freezer.py │ │ ├── functional.py │ │ ├── init.py │ │ └── modules │ │ │ ├── __init__.py │ │ │ ├── conv.py │ │ │ ├── linear.py │ │ │ └── mlp.py │ └── setup.py │ └── pointnet2.py ├── requirements.txt ├── test.py ├── test_file ├── real_data │ ├── 0000_cloud.pcd │ └── 0001_cloud.pcd ├── real_data_predict │ ├── 0000_cloud.p │ └── 0001_cloud.p ├── virtual_data │ ├── 00001_view_1.p │ └── 2946_view_0_noise.p └── virtual_data_predict │ ├── 00001_view_1.p │ └── 2946_view_0_noise.p ├── train.py ├── utils.py └── vis ├── vis_grasp.py └── visualization_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/README.md -------------------------------------------------------------------------------- /count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/count.py -------------------------------------------------------------------------------- /dataset_utils/eval_score/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/dataset_utils/eval_score/configs/config.py -------------------------------------------------------------------------------- /dataset_utils/eval_score/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/dataset_utils/eval_score/eval.py -------------------------------------------------------------------------------- /dataset_utils/eval_score/eval_utils/evaluation_data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/dataset_utils/eval_score/eval_utils/evaluation_data_generator.py -------------------------------------------------------------------------------- /dataset_utils/eval_score/eval_utils/pointcloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/dataset_utils/eval_score/eval_utils/pointcloud.py -------------------------------------------------------------------------------- /dataset_utils/eval_score/eval_utils/torch_scene_point_cloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/dataset_utils/eval_score/eval_utils/torch_scene_point_cloud.py -------------------------------------------------------------------------------- /dataset_utils/get_regiondataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/dataset_utils/get_regiondataset.py -------------------------------------------------------------------------------- /dataset_utils/scoredataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/dataset_utils/scoredataset.py -------------------------------------------------------------------------------- /markdown/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/1.png -------------------------------------------------------------------------------- /markdown/1_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/1_1.png -------------------------------------------------------------------------------- /markdown/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/2.png -------------------------------------------------------------------------------- /markdown/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/3.png -------------------------------------------------------------------------------- /markdown/4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/4.png -------------------------------------------------------------------------------- /markdown/4_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/4_2.png -------------------------------------------------------------------------------- /markdown/score2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/score2.png -------------------------------------------------------------------------------- /markdown/score3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/score3.png -------------------------------------------------------------------------------- /markdown/vgr2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/vgr2.png -------------------------------------------------------------------------------- /markdown/vgr3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/markdown/vgr3.png -------------------------------------------------------------------------------- /multi_model/gripper_region_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/gripper_region_network.py -------------------------------------------------------------------------------- /multi_model/score_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/score_network.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/csrc/ball_query.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/csrc/ball_query.h -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/csrc/ball_query_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/csrc/ball_query_kernel.cu -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/csrc/grouping.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/csrc/grouping.h -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/csrc/grouping_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/csrc/grouping_kernel.cu -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/csrc/interpolate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/csrc/interpolate.h -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/csrc/interpolate_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/csrc/interpolate_kernel.cu -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/csrc/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/csrc/main.cpp -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/csrc/sampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/csrc/sampling.h -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/csrc/sampling_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/csrc/sampling_kernel.cu -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/function.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/functions/csrc/gather_knn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/functions/csrc/gather_knn.h -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/functions/csrc/gather_knn_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/functions/csrc/gather_knn_kernel.cu -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/functions/csrc/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/functions/csrc/main.cpp -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/functions/gather_knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/functions/gather_knn.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/functions/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/functions/setup.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/modules.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/nn/__init__.py: -------------------------------------------------------------------------------- 1 | from .modules import * 2 | -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/nn/freezer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/nn/freezer.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/nn/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/nn/functional.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/nn/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/nn/init.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/nn/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/nn/modules/__init__.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/nn/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/nn/modules/conv.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/nn/modules/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/nn/modules/linear.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/nn/modules/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/nn/modules/mlp.py -------------------------------------------------------------------------------- /multi_model/utils/pn2_utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pn2_utils/setup.py -------------------------------------------------------------------------------- /multi_model/utils/pointnet2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/multi_model/utils/pointnet2.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/test.py -------------------------------------------------------------------------------- /test_file/real_data/0000_cloud.pcd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/test_file/real_data/0000_cloud.pcd -------------------------------------------------------------------------------- /test_file/real_data/0001_cloud.pcd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/test_file/real_data/0001_cloud.pcd -------------------------------------------------------------------------------- /test_file/real_data_predict/0000_cloud.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/test_file/real_data_predict/0000_cloud.p -------------------------------------------------------------------------------- /test_file/real_data_predict/0001_cloud.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/test_file/real_data_predict/0001_cloud.p -------------------------------------------------------------------------------- /test_file/virtual_data/00001_view_1.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/test_file/virtual_data/00001_view_1.p -------------------------------------------------------------------------------- /test_file/virtual_data/2946_view_0_noise.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/test_file/virtual_data/2946_view_0_noise.p -------------------------------------------------------------------------------- /test_file/virtual_data_predict/00001_view_1.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/test_file/virtual_data_predict/00001_view_1.p -------------------------------------------------------------------------------- /test_file/virtual_data_predict/2946_view_0_noise.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/test_file/virtual_data_predict/2946_view_0_noise.p -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/utils.py -------------------------------------------------------------------------------- /vis/vis_grasp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/vis/vis_grasp.py -------------------------------------------------------------------------------- /vis/visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaobinglei/REGNet_for_3D_Grasping/HEAD/vis/visualization_utils.py --------------------------------------------------------------------------------