Deep Bayesian-Assisted Keypoint Detection for Pose Estimation in Assembly Automation

Pose estimation is crucial for automating assembly tasks, yet achieving sufficient accuracy for assembly automation remains challenging and part-specific. This paper presents a novel, streamlined approach to pose estimation that facilitates automation of assembly tasks. Our proposed method employs d...

Full description

Bibliographic Details
Main Authors: Debo Shi, Alireza Rahimpour, Amin Ghafourian, Mohammad Mahdi Naddaf Shargh, Devesh Upadhyay, Ty A. Lasky, Iman Soltani
Format: Article
Language:English
Published: MDPI AG 2023-07-01
Series:Sensors
Subjects:
Online Access:https://www.mdpi.com/1424-8220/23/13/6107
_version_ 1797436001835024384
author Debo Shi
Alireza Rahimpour
Amin Ghafourian
Mohammad Mahdi Naddaf Shargh
Devesh Upadhyay
Ty A. Lasky
Iman Soltani
author_facet Debo Shi
Alireza Rahimpour
Amin Ghafourian
Mohammad Mahdi Naddaf Shargh
Devesh Upadhyay
Ty A. Lasky
Iman Soltani
author_sort Debo Shi
collection DOAJ
description Pose estimation is crucial for automating assembly tasks, yet achieving sufficient accuracy for assembly automation remains challenging and part-specific. This paper presents a novel, streamlined approach to pose estimation that facilitates automation of assembly tasks. Our proposed method employs deep learning on a limited number of annotated images to identify a set of keypoints on the parts of interest. To compensate for network shortcomings and enhance accuracy we incorporated a Bayesian updating stage that leverages our detailed knowledge of the assembly part design. This Bayesian updating step refines the network output, significantly improving pose estimation accuracy. For this purpose, we utilized a subset of network-generated keypoint positions with higher quality as measurements, while for the remaining keypoints, the network outputs only serve as priors. The geometry data aid in constructing likelihood functions, which in turn result in enhanced posterior distributions of keypoint pixel positions. We then employed the maximum a posteriori (MAP) estimates of keypoint locations to obtain a final pose, allowing for an update to the nominal assembly trajectory. We evaluated our method on a 14-point snap-fit dash trim assembly for a Ford Mustang dashboard, demonstrating promising results. Our approach does not require tailoring to new applications, nor does it rely on extensive machine learning expertise or large amounts of training data. This makes our method a scalable and adaptable solution for the production floors.
first_indexed 2024-03-09T10:56:23Z
format Article
id doaj.art-830442d115544f3e905c367a61acb620
institution Directory Open Access Journal
issn 1424-8220
language English
last_indexed 2024-03-09T10:56:23Z
publishDate 2023-07-01
publisher MDPI AG
record_format Article
series Sensors
spelling doaj.art-830442d115544f3e905c367a61acb6202023-12-01T01:37:24ZengMDPI AGSensors1424-82202023-07-012313610710.3390/s23136107Deep Bayesian-Assisted Keypoint Detection for Pose Estimation in Assembly AutomationDebo Shi0Alireza Rahimpour1Amin Ghafourian2Mohammad Mahdi Naddaf Shargh3Devesh Upadhyay4Ty A. Lasky5Iman Soltani6Department of Electrical and Computer Engineering, University of California Davis, Davis, CA 95616, USAGreenfield Labs, Ford Motor Company, Palo Alto, CA 94304, USADepartment of Mechanical and Aerospace Engineering, University of California Davis, Davis, CA 95616, USAGreenfield Labs, Ford Motor Company, Palo Alto, CA 94304, USAGreenfield Labs, Ford Motor Company, Palo Alto, CA 94304, USADepartment of Mechanical and Aerospace Engineering, University of California Davis, Davis, CA 95616, USADepartment of Mechanical and Aerospace Engineering, University of California Davis, Davis, CA 95616, USAPose estimation is crucial for automating assembly tasks, yet achieving sufficient accuracy for assembly automation remains challenging and part-specific. This paper presents a novel, streamlined approach to pose estimation that facilitates automation of assembly tasks. Our proposed method employs deep learning on a limited number of annotated images to identify a set of keypoints on the parts of interest. To compensate for network shortcomings and enhance accuracy we incorporated a Bayesian updating stage that leverages our detailed knowledge of the assembly part design. This Bayesian updating step refines the network output, significantly improving pose estimation accuracy. For this purpose, we utilized a subset of network-generated keypoint positions with higher quality as measurements, while for the remaining keypoints, the network outputs only serve as priors. The geometry data aid in constructing likelihood functions, which in turn result in enhanced posterior distributions of keypoint pixel positions. We then employed the maximum a posteriori (MAP) estimates of keypoint locations to obtain a final pose, allowing for an update to the nominal assembly trajectory. We evaluated our method on a 14-point snap-fit dash trim assembly for a Ford Mustang dashboard, demonstrating promising results. Our approach does not require tailoring to new applications, nor does it rely on extensive machine learning expertise or large amounts of training data. This makes our method a scalable and adaptable solution for the production floors.https://www.mdpi.com/1424-8220/23/13/6107keypoint detectionpose estimationassembly automationmanufacturing automationdeep learningAI
spellingShingle Debo Shi
Alireza Rahimpour
Amin Ghafourian
Mohammad Mahdi Naddaf Shargh
Devesh Upadhyay
Ty A. Lasky
Iman Soltani
Deep Bayesian-Assisted Keypoint Detection for Pose Estimation in Assembly Automation
Sensors
keypoint detection
pose estimation
assembly automation
manufacturing automation
deep learning
AI
title Deep Bayesian-Assisted Keypoint Detection for Pose Estimation in Assembly Automation
title_full Deep Bayesian-Assisted Keypoint Detection for Pose Estimation in Assembly Automation
title_fullStr Deep Bayesian-Assisted Keypoint Detection for Pose Estimation in Assembly Automation
title_full_unstemmed Deep Bayesian-Assisted Keypoint Detection for Pose Estimation in Assembly Automation
title_short Deep Bayesian-Assisted Keypoint Detection for Pose Estimation in Assembly Automation
title_sort deep bayesian assisted keypoint detection for pose estimation in assembly automation
topic keypoint detection
pose estimation
assembly automation
manufacturing automation
deep learning
AI
url https://www.mdpi.com/1424-8220/23/13/6107
work_keys_str_mv AT deboshi deepbayesianassistedkeypointdetectionforposeestimationinassemblyautomation
AT alirezarahimpour deepbayesianassistedkeypointdetectionforposeestimationinassemblyautomation
AT aminghafourian deepbayesianassistedkeypointdetectionforposeestimationinassemblyautomation
AT mohammadmahdinaddafshargh deepbayesianassistedkeypointdetectionforposeestimationinassemblyautomation
AT deveshupadhyay deepbayesianassistedkeypointdetectionforposeestimationinassemblyautomation
AT tyalasky deepbayesianassistedkeypointdetectionforposeestimationinassemblyautomation
AT imansoltani deepbayesianassistedkeypointdetectionforposeestimationinassemblyautomation