高级检索

基于空间曲线卷积神经网络的CAD模型缝隙识别方法

Seam recognition in CAD models via spatial curve convolutional neural networks

  • 摘要: 在CAD模型中, 需要进行由于曲面缝隙引起的缺陷识别. 针对传统的基于人工特征与几何距离阈值的识别方法在多尺度适应性和拓扑判别精度上存在显著局限, 难以满足复杂工业场景的检测要求的问题, 采用数据驱动的思路, 融合几何形变增强与深度特征学习技术, 提出一种基于空间曲线卷积算子的神经网络的CAD模型缝隙识别方法. 首先设计基于径向基函数的保距数据增强算法, 依托局部刚性约束变换生成基于原始样本的高质量合成样本, 缓解小样本训练场景下的模型泛化性能瓶颈; 然后构建融合多级卷积模块和多头自注意力机制的网络, 通过动态强化缝隙特征的几何显著性及长程依赖关系实现端到端的高精度缺陷识别. 将人工构建的371个原始样本扩增至4 081个, 并将其随机划分为训练集和测试集; 模型在含有2 860个样本的训练集上完成, 训练结束后, 在剩余的测试集中开展模型性能测试. 实验结果表明, 所提方法的准确率达到97.87%, 平均交并比为96.42%, 测试总费时0.26 s, 综合性能显著优于传统规则化方法, 为CAD模型智能几何清理提供了高效可靠的技术路径, 该方法可以拓展至其他复杂的几何缺陷识别场景.

     

    Abstract: In CAD models, defect recognition arising from surface gaps is essential. Traditional methods, which rely on manual feature extraction and geometric distance thresholds, exhibit notable limitations in multi-scale adaptability and topological discrimination accuracy, making it difficult to meet the demands of complex industrial inspection. To address this issue, a data-driven approach integrating geometric deformation augmentation and deep feature learning is adopted, and a neural network-based gap recognition method using spatial curve convolution operators is proposed. First, a distance-preserving data augmentation algorithm based on radial basis functions was designed. By applying locally rigid transformation constraints, high-quality synthetic samples were generated from the original dataset, alleviating the model generalization bottleneck in small-sample training scenarios. Then, a network incorporating multi-level convolution modules and multi-head self-attention mechanisms was constructed. This network dynamically enhances the geometric salience of gap features and captures long-range dependencies, enabling end-to-end high-precision defect recognition. The original 371 manually constructed samples were expanded to 4 081, which were then randomly divided into training and test sets. The model was trained on a set of 2,860 samples, and after training, performance was evaluated on the remaining test set. Experimental results demonstrate that the proposed method achieves an accuracy of 97.87% and a mean Intersection over Union (mIoU) of 96.42%, with a total testing time of 0.26 s. Its overall performance is significantly superior to that of traditional rule-based methods. This approach provides an efficient and reliable technical solution for intelligent geometric cleaning of CAD models and can be extended to other complex geometric defect recognition scenarios.

     

/

返回文章
返回