Engine Assembly Dataset
Engine Assembly Dataset including 8 classes of object and targets. The dataset was generated by capturing 195 real images and annotating them with the correct labels (bounding boxes, keypoints and polygons). Further augmentation of the images completes the dataset to around 280,000 images. All tools and scripts to generate and replicate the dataset are provided, as well as the trained model and visualization scripts.
Link: https://zenodo.org/records/7669593
6DAPose – Synthetic Assembly Pose Dataset
6DAPose dataset contains an object assembly in an ordered manner following its assembly steps. Each dataset contains 431 simulated view samples of the assembly for each assembly step. For each view sample, RGB and Depth Image, Segmentation map, ground truth values for 6D object pose and camera pose and model information are recorded following the BOP format . There are two datasets representing a 3-step fidget gear assembly and a 4-step Nema17 gear reducer assembly.
Link: https://zenodo.org/records/10117869
Planetary Gear Dataset
Planetary Gear Dataset including 6 classes of object. The dataset was generated by capturing 150 real images (Intel Realsense D435) and annotating them with the correct labels (bounding boxes, keypoints and polygons). Further augmentation of the images completes the dataset to around 170,000 images. All tools and scripts to generate and replicate the dataset are provided, as well as the trained model and visualization scripts.