训练图像对的扩增方法、装置、存储介质及电子装置制造方法及图纸

本申请公开了一种训练图像对的扩增方法、装置、存储介质及电子装置。该训练图像对的扩增方法包括：将人工标注图像对成对输入GAN，GAN学会标注图像到真实图像的转换规则；利用BIM模型生成大量的自动标注图像；将自动标注图像输入GAN，GAN根据转换规则合成与所述自动标注图像对应的真实图像；根据自动标注图像和对应的真实图像生成自动标注图像对；组合自动标注图像对和人工标注图像对，得到训练图像对集。本申请解决了手动标注工作效率低，且人力成本高的技术问题。成本高的技术问题。成本高的技术问题。

【技术实现步骤摘要】
训练图像对的扩增方法、装置、存储介质及电子装置


[0001]本申请涉及建筑工程图像标注领域，具体而言，涉及一种训练图像对的扩增方法、装置、存储介质及电子装置。

技术介绍

[0002]专利技术人发现，针对建筑工程领域中特定的工作环境，需要收集相关图像数据集并进行手动标注。而这必然导致大量的图像标注都需要手动来完成，工作效率低，人力成本高。
[0003]针对相关技术中手动标注工作效率低，且人力成本高的问题，目前尚未提出有效的解决方案。

技术实现思路

[0004]本申请的主要目的在于提供一种训练图像对的扩增方法、装置、存储介质及电子装置，以解决手动标注工作效率低，人力成本高，且对人员专业要求高的问题。
[0005]为了实现上述目的，根据本申请的一个方面，提供了一种训练图像对的扩增方法、装置、存储介质及电子装置。
[0006]根据本申请的训练图像对的扩增方法包括：将人工标注图像对成对输入GAN， GAN学会标注图像到真实图像的转换规则；利用BIM模型生成大量的自动标注图像；将自动标注图像输入GAN，GAN根据转换本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种训练图像对的扩增方法，其特征在于，包括：将人工标注图像对成对输入GAN，GAN学会标注图像到真实图像的转换规则；利用BIM模型生成自动标注图像；将自动标注图像输入GAN，GAN根据转换规则合成与所述自动标注图像对应的真实图像；根据自动标注图像和对应的真实图像生成自动标注图像对；组合自动标注图像对和人工标注图像对，得到训练图像对集。2.根据权利要求1所述的训练图像对的扩增方法，其特征在于，将人工标注图像对成对输入GAN，GAN学会标注图像到真实图像的映射规则之前还包括：通过对拍摄的建筑工程图像进行人工标注，得到人工标注图像；根据人工标注图像和对应的建筑工程图像生成人工标注图像对。3.根据权利要求1所述的训练图像对的扩增方法，其特征在于，利用BIM模型生成自动标注图像包括：调整BIM模型中构建的颜色，使BIM模型构建的颜色与标注图像中对应构件的颜色一致；利用漫游或录屏功能自动生成大量的自动标注图像。4.根据权利要求1所述的训练图像对的扩增方法，其特征在于，将人工标注图像对成对输入GAN，GAN学会标注图像到真实图像的转换规则包括：将人工标注图像对中的人工标注图像输入GAN的生成器中，生成一合成图像；将人工标注图像对中的真实图像和合成图像输入GAN的判别器中，判断两者的相似度；其中，所述生成器和所述判别器为对抗关系。5.一种训练图像对的扩增装置，其特征在于，包括：训练模块，用于将人工标注图像对成对输入GAN，GAN学会标注图像到真实图像的转换规则；自动标注模块，用于利用BIM模型生成...

【专利技术属性】
技术研发人员：苏星，李路平，周筱璇，范佳敏，
申请(专利权)人：浙江大学，
类型：发明
国别省市：