├── .gitignore ├── README.md ├── configs ├── dsqnet_config.yml ├── segnet_config.yml └── sqnet_config.yml ├── data_generation.py ├── evaluation.py ├── figures └── pipeline.png ├── functions ├── RenderOption.json ├── data_generator.py ├── data_reader.py ├── iou_calculator.py ├── object_class.py ├── pc_preprocess.py ├── primitives.py ├── utils_datagen.py ├── utils_numpy.py └── utils_torch.py ├── grasp_pose_generation.py ├── grasping ├── assets │ ├── finger.ply │ ├── hand.ply │ ├── link6.ply │ └── link7.ply ├── grasp_points_sampler.py └── gripper.py ├── loader ├── __init__.py ├── object_dataset.py └── primitive_dataset.py ├── loss ├── __init__.py ├── dsq_loss.py ├── segmentation_loss.py └── sq_loss.py ├── metrics ├── __init__.py ├── meter.py ├── segmentation_metric.py └── volumetric_iou.py ├── models ├── __init__.py ├── dgcnn.py ├── dsqnet.py ├── segmenation_network.py └── sqnet.py ├── object_params ├── bottle_cone.json ├── box.json ├── cone.json ├── cup_with_lid.json ├── cylinder.json ├── dumbbell.json ├── ellipsoid.json ├── hammer_cylinder.json ├── padlock.json ├── screw_driver.json ├── truncated_cone.json └── truncated_torus.json ├── optimizers.py ├── requirements.txt ├── train.py ├── trainers ├── __init__.py ├── recognition_trainer.py └── segmentation_trainer.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/README.md -------------------------------------------------------------------------------- /configs/dsqnet_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/configs/dsqnet_config.yml -------------------------------------------------------------------------------- /configs/segnet_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/configs/segnet_config.yml -------------------------------------------------------------------------------- /configs/sqnet_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/configs/sqnet_config.yml -------------------------------------------------------------------------------- /data_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/data_generation.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/evaluation.py -------------------------------------------------------------------------------- /figures/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/figures/pipeline.png -------------------------------------------------------------------------------- /functions/RenderOption.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/RenderOption.json -------------------------------------------------------------------------------- /functions/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/data_generator.py -------------------------------------------------------------------------------- /functions/data_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/data_reader.py -------------------------------------------------------------------------------- /functions/iou_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/iou_calculator.py -------------------------------------------------------------------------------- /functions/object_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/object_class.py -------------------------------------------------------------------------------- /functions/pc_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/pc_preprocess.py -------------------------------------------------------------------------------- /functions/primitives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/primitives.py -------------------------------------------------------------------------------- /functions/utils_datagen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/utils_datagen.py -------------------------------------------------------------------------------- /functions/utils_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/utils_numpy.py -------------------------------------------------------------------------------- /functions/utils_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/functions/utils_torch.py -------------------------------------------------------------------------------- /grasp_pose_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/grasp_pose_generation.py -------------------------------------------------------------------------------- /grasping/assets/finger.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/grasping/assets/finger.ply -------------------------------------------------------------------------------- /grasping/assets/hand.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/grasping/assets/hand.ply -------------------------------------------------------------------------------- /grasping/assets/link6.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/grasping/assets/link6.ply -------------------------------------------------------------------------------- /grasping/assets/link7.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/grasping/assets/link7.ply -------------------------------------------------------------------------------- /grasping/grasp_points_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/grasping/grasp_points_sampler.py -------------------------------------------------------------------------------- /grasping/gripper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/grasping/gripper.py -------------------------------------------------------------------------------- /loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/loader/__init__.py -------------------------------------------------------------------------------- /loader/object_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/loader/object_dataset.py -------------------------------------------------------------------------------- /loader/primitive_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/loader/primitive_dataset.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/dsq_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/loss/dsq_loss.py -------------------------------------------------------------------------------- /loss/segmentation_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/loss/segmentation_loss.py -------------------------------------------------------------------------------- /loss/sq_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/loss/sq_loss.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /metrics/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/metrics/meter.py -------------------------------------------------------------------------------- /metrics/segmentation_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/metrics/segmentation_metric.py -------------------------------------------------------------------------------- /metrics/volumetric_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/metrics/volumetric_iou.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/dgcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/models/dgcnn.py -------------------------------------------------------------------------------- /models/dsqnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/models/dsqnet.py -------------------------------------------------------------------------------- /models/segmenation_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/models/segmenation_network.py -------------------------------------------------------------------------------- /models/sqnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/models/sqnet.py -------------------------------------------------------------------------------- /object_params/bottle_cone.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/bottle_cone.json -------------------------------------------------------------------------------- /object_params/box.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/box.json -------------------------------------------------------------------------------- /object_params/cone.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/cone.json -------------------------------------------------------------------------------- /object_params/cup_with_lid.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/cup_with_lid.json -------------------------------------------------------------------------------- /object_params/cylinder.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/cylinder.json -------------------------------------------------------------------------------- /object_params/dumbbell.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/dumbbell.json -------------------------------------------------------------------------------- /object_params/ellipsoid.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/ellipsoid.json -------------------------------------------------------------------------------- /object_params/hammer_cylinder.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/hammer_cylinder.json -------------------------------------------------------------------------------- /object_params/padlock.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/padlock.json -------------------------------------------------------------------------------- /object_params/screw_driver.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/screw_driver.json -------------------------------------------------------------------------------- /object_params/truncated_cone.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/truncated_cone.json -------------------------------------------------------------------------------- /object_params/truncated_torus.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/object_params/truncated_torus.json -------------------------------------------------------------------------------- /optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/optimizers.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/train.py -------------------------------------------------------------------------------- /trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/trainers/__init__.py -------------------------------------------------------------------------------- /trainers/recognition_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/trainers/recognition_trainer.py -------------------------------------------------------------------------------- /trainers/segmentation_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/trainers/segmentation_trainer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seungyeon-k/DSQNet-public/HEAD/utils.py --------------------------------------------------------------------------------