基于小样本的工程图纸识别模型的训练方法及装置

本发明涉及图像处理技术领域，公开了一种基于小样本的工程图纸识别模型的训练方法及装置，该方法包括：对工程图纸集合分类得到多个图纸样本组合；分析每一图纸样本组合得到图纸分析结果；根据所有图纸分析结果，筛选出满足小样本类别条件的第一图纸样本组合；将第一图纸样本组合输入至自编码结构模型中进行重构得到第二图纸样本组合，并将第一图纸样本组合及第一图纸样本组合作为新的第一图纸样本组合，对工程图纸集合进行更新，得到目标工程图纸集合；将目标工程图纸集合输入至机器学习模型中进行训练，得到工程图纸识别模型。可见，实施本发明能够提高工程图纸识别模型的训练准确性，有利于提高工程图纸识别模型的识别准确性及识别效率。