├── .gitignore ├── LICENSE ├── README.md ├── configs ├── hyp.single.yaml └── linemod │ ├── ape.yaml │ ├── benchvise.yaml │ ├── cam.yaml │ ├── can.yaml │ ├── cat.yaml │ ├── cube.yaml │ ├── driller.yaml │ ├── duck.yaml │ ├── eggbox.yaml │ ├── glue.yaml │ ├── holepuncher.yaml │ ├── iron.yaml │ ├── lamp.yaml │ ├── linemod_camera.json │ └── phone.yaml ├── data_curation ├── CreateArucoBoard.ipynb ├── CreatingDataset.md ├── calibrate.ipynb ├── convert_linemod.py ├── create_data_splits.ipynb ├── create_dataset.ipynb └── pose_utils.py ├── detect.py ├── docker ├── hubconf.py ├── models ├── __init__.py ├── common.py ├── experimental.py ├── export.py ├── hub │ ├── anchors.yaml │ ├── yolov3-spp.yaml │ ├── yolov3-tiny.yaml │ ├── yolov3.yaml │ ├── yolov5-fpn.yaml │ ├── yolov5-p2.yaml │ ├── yolov5-p6.yaml │ ├── yolov5-p7.yaml │ ├── yolov5-panet.yaml │ ├── yolov5l6.yaml │ ├── yolov5m6.yaml │ ├── yolov5s6.yaml │ └── yolov5x6.yaml ├── yolo.py ├── yolov5l_6dpose_bifpn.yaml ├── yolov5s_6dpose_bifpn.yaml └── yolov5x_6dpose_bifpn.yaml ├── requirements.txt ├── results └── images │ ├── acc_vs_speed.png │ ├── cat.gif │ ├── cube.gif │ ├── example_cat.jpg │ ├── glue.gif │ ├── iron.gif │ ├── lamp.gif │ ├── multi.png │ ├── phantom.gif │ ├── screw.gif │ └── screw_train.gif ├── test.py ├── train.py ├── tutorial.ipynb └── utils ├── __init__.py ├── activations.py ├── autoanchor.py ├── calc_min_distances.h ├── compute_overlap.pyx ├── datasets.py ├── downloads.py ├── general.py ├── google_utils.py ├── image.py ├── loss.py ├── metrics.py ├── occlude.py ├── plots.py ├── pose_utils.py ├── setup.py ├── torch_utils.py └── wandb_logging ├── __init__.py ├── log_dataset.py └── wandb_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/README.md -------------------------------------------------------------------------------- /configs/hyp.single.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/hyp.single.yaml -------------------------------------------------------------------------------- /configs/linemod/ape.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/ape.yaml -------------------------------------------------------------------------------- /configs/linemod/benchvise.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/benchvise.yaml -------------------------------------------------------------------------------- /configs/linemod/cam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/cam.yaml -------------------------------------------------------------------------------- /configs/linemod/can.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/can.yaml -------------------------------------------------------------------------------- /configs/linemod/cat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/cat.yaml -------------------------------------------------------------------------------- /configs/linemod/cube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/cube.yaml -------------------------------------------------------------------------------- /configs/linemod/driller.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/driller.yaml -------------------------------------------------------------------------------- /configs/linemod/duck.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/duck.yaml -------------------------------------------------------------------------------- /configs/linemod/eggbox.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/eggbox.yaml -------------------------------------------------------------------------------- /configs/linemod/glue.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/glue.yaml -------------------------------------------------------------------------------- /configs/linemod/holepuncher.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/holepuncher.yaml -------------------------------------------------------------------------------- /configs/linemod/iron.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/iron.yaml -------------------------------------------------------------------------------- /configs/linemod/lamp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/lamp.yaml -------------------------------------------------------------------------------- /configs/linemod/linemod_camera.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/linemod_camera.json -------------------------------------------------------------------------------- /configs/linemod/phone.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/configs/linemod/phone.yaml -------------------------------------------------------------------------------- /data_curation/CreateArucoBoard.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/data_curation/CreateArucoBoard.ipynb -------------------------------------------------------------------------------- /data_curation/CreatingDataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/data_curation/CreatingDataset.md -------------------------------------------------------------------------------- /data_curation/calibrate.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/data_curation/calibrate.ipynb -------------------------------------------------------------------------------- /data_curation/convert_linemod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/data_curation/convert_linemod.py -------------------------------------------------------------------------------- /data_curation/create_data_splits.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/data_curation/create_data_splits.ipynb -------------------------------------------------------------------------------- /data_curation/create_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/data_curation/create_dataset.ipynb -------------------------------------------------------------------------------- /data_curation/pose_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/data_curation/pose_utils.py -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/detect.py -------------------------------------------------------------------------------- /docker: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/docker -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/hubconf.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/export.py -------------------------------------------------------------------------------- /models/hub/anchors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/anchors.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov3-tiny.yaml -------------------------------------------------------------------------------- /models/hub/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov3.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov5-p2.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov5-p6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov5-p7.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /models/hub/yolov5l6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov5l6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5m6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov5m6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov5s6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5x6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/hub/yolov5x6.yaml -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov5l_6dpose_bifpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/yolov5l_6dpose_bifpn.yaml -------------------------------------------------------------------------------- /models/yolov5s_6dpose_bifpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/yolov5s_6dpose_bifpn.yaml -------------------------------------------------------------------------------- /models/yolov5x_6dpose_bifpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/models/yolov5x_6dpose_bifpn.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/images/acc_vs_speed.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/acc_vs_speed.png -------------------------------------------------------------------------------- /results/images/cat.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/cat.gif -------------------------------------------------------------------------------- /results/images/cube.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/cube.gif -------------------------------------------------------------------------------- /results/images/example_cat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/example_cat.jpg -------------------------------------------------------------------------------- /results/images/glue.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/glue.gif -------------------------------------------------------------------------------- /results/images/iron.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/iron.gif -------------------------------------------------------------------------------- /results/images/lamp.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/lamp.gif -------------------------------------------------------------------------------- /results/images/multi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/multi.png -------------------------------------------------------------------------------- /results/images/phantom.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/phantom.gif -------------------------------------------------------------------------------- /results/images/screw.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/screw.gif -------------------------------------------------------------------------------- /results/images/screw_train.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/results/images/screw_train.gif -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/train.py -------------------------------------------------------------------------------- /tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/tutorial.ipynb -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/calc_min_distances.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/calc_min_distances.h -------------------------------------------------------------------------------- /utils/compute_overlap.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/compute_overlap.pyx -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/downloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/downloads.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/image.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/occlude.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/occlude.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/pose_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/pose_utils.py -------------------------------------------------------------------------------- /utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/setup.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/wandb_logging/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/wandb_logging/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/wandb_logging/log_dataset.py -------------------------------------------------------------------------------- /utils/wandb_logging/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cviviers/YOLOv5-6D-Pose/HEAD/utils/wandb_logging/wandb_utils.py --------------------------------------------------------------------------------