├── .gitignore ├── .vscode └── settings.json ├── LICENSE ├── README.md ├── Utils.py ├── blender_dataset_generator.py ├── blender_main.py ├── config.yml ├── data_augmentation.py ├── dataset_info.yml ├── datasets.py ├── docker ├── dockerfile └── run_container.sh ├── eval_ycb.py ├── eval_ycbineoat.py ├── media ├── 0000000rgbA.png ├── 0000000rgbB.png ├── cup_stacking_and_charger.gif ├── curves.jpg ├── eoat1.jpg ├── eoat2.jpg ├── industrial_insertion.gif ├── manipulation1.gif ├── occlusion.gif ├── syndata_gen.gif ├── ycb_packing.gif ├── ycb_results.jpg ├── ycbineoat_benchmark.jpg └── youtube_thumbnail.jpg ├── network_modules.py ├── object_models └── bunny │ ├── 00845f.png │ ├── 1.mtl │ ├── 1.obj │ └── 1.ply ├── offscreen_renderer.py ├── predict.py ├── predict_ros.py ├── problems.py ├── produce_train_pair_data.py ├── se3_tracknet.py ├── train.py ├── train.sh └── vispy_renderer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/README.md -------------------------------------------------------------------------------- /Utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/Utils.py -------------------------------------------------------------------------------- /blender_dataset_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/blender_dataset_generator.py -------------------------------------------------------------------------------- /blender_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/blender_main.py -------------------------------------------------------------------------------- /config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/config.yml -------------------------------------------------------------------------------- /data_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/data_augmentation.py -------------------------------------------------------------------------------- /dataset_info.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/dataset_info.yml -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/datasets.py -------------------------------------------------------------------------------- /docker/dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/docker/dockerfile -------------------------------------------------------------------------------- /docker/run_container.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/docker/run_container.sh -------------------------------------------------------------------------------- /eval_ycb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/eval_ycb.py -------------------------------------------------------------------------------- /eval_ycbineoat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/eval_ycbineoat.py -------------------------------------------------------------------------------- /media/0000000rgbA.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/0000000rgbA.png -------------------------------------------------------------------------------- /media/0000000rgbB.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/0000000rgbB.png -------------------------------------------------------------------------------- /media/cup_stacking_and_charger.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/cup_stacking_and_charger.gif -------------------------------------------------------------------------------- /media/curves.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/curves.jpg -------------------------------------------------------------------------------- /media/eoat1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/eoat1.jpg -------------------------------------------------------------------------------- /media/eoat2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/eoat2.jpg -------------------------------------------------------------------------------- /media/industrial_insertion.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/industrial_insertion.gif -------------------------------------------------------------------------------- /media/manipulation1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/manipulation1.gif -------------------------------------------------------------------------------- /media/occlusion.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/occlusion.gif -------------------------------------------------------------------------------- /media/syndata_gen.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/syndata_gen.gif -------------------------------------------------------------------------------- /media/ycb_packing.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/ycb_packing.gif -------------------------------------------------------------------------------- /media/ycb_results.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/ycb_results.jpg -------------------------------------------------------------------------------- /media/ycbineoat_benchmark.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/ycbineoat_benchmark.jpg -------------------------------------------------------------------------------- /media/youtube_thumbnail.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/media/youtube_thumbnail.jpg -------------------------------------------------------------------------------- /network_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/network_modules.py -------------------------------------------------------------------------------- /object_models/bunny/00845f.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/object_models/bunny/00845f.png -------------------------------------------------------------------------------- /object_models/bunny/1.mtl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/object_models/bunny/1.mtl -------------------------------------------------------------------------------- /object_models/bunny/1.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/object_models/bunny/1.obj -------------------------------------------------------------------------------- /object_models/bunny/1.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/object_models/bunny/1.ply -------------------------------------------------------------------------------- /offscreen_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/offscreen_renderer.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/predict.py -------------------------------------------------------------------------------- /predict_ros.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/predict_ros.py -------------------------------------------------------------------------------- /problems.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/problems.py -------------------------------------------------------------------------------- /produce_train_pair_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/produce_train_pair_data.py -------------------------------------------------------------------------------- /se3_tracknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/se3_tracknet.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | export CUDA_VISIBLE_DEVICES=0 4 | python3 train.py -------------------------------------------------------------------------------- /vispy_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenbowen123/iros20-6d-pose-tracking/HEAD/vispy_renderer.py --------------------------------------------------------------------------------