用CNN检测物体角落点,提升机械臂视觉伺服精度
Keypoint Detection Technique for Image-Based Visual Servoing of Manipulators
- 基于VGG-19改进的CNN模型,专注识别物体角点特征
- 通过数据增强与平均池化,验证损失降低50%
- 适合需要高精度视觉伺服的机器人研发人员
本文提出一种基于卷积神经网络(CNN)的关键点检测技术,以提升现有深度视觉伺服(DVS)模型性能。为验证图像基视觉伺服(IBVS)算法的收敛性,先使用标定标记在真实场景中进行特征检测实验。为克服标定标记的局限性,新检测器聚焦于提取更接近真实物体的角点特征。利用安装在机械臂末端的摄像头在任务空间随机运动时采集样本,自动标注并结合翻转与旋转扩充数据集。基于ImageNet预训练的VGG-19模型进行改造:固定基础层权重,仅更新全连接层权重,以最小化预测值与真实角点像素坐标间的平均绝对误差。模型经两项改进:将最大池化替换为平均池化,并采用随训练周期递减的自适应学习率,使验证损失降低50%。最后通过k折交叉验证评估模型可靠性。
原文摘要 · Abstract (English)
This paper introduces an innovative keypoint detection technique based on Convolutional Neural Networks (CNNs) to enhance the performance of existing Deep Visual Servoing (DVS) models. To validate the convergence of the Image-Based Visual Servoing (IBVS) algorithm, real-world experiments utilizing fiducial markers for feature detection are conducted before designing the CNN-based feature detector. To address the limitations of fiducial markers, the novel feature detector focuses on extracting keypoints that represent the corners of a more realistic object compared to fiducial markers. A dataset is generated from sample data captured by the camera mounted on the robot end-effector while the robot operates randomly in the task space. The samples are automatically labeled, and the dataset size is increased by flipping and rotation. The CNN model is developed by modifying the VGG-19 pre-trained on the ImageNet dataset. While the weights in the base model remain fixed, the fully connected layer's weights are updated to minimize the mean absolute error, defined based on the deviation of predictions from the real pixel coordinates of the corners. The model undergoes two modifications: replacing max-pooling with average-pooling in the base model and implementing an adaptive learning rate that decreases during epochs. These changes lead to a 50 percent reduction in validation loss. Finally, the trained model's reliability is assessed through k-fold cross-validation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。