北京有道数智科技有限公司

本发明属于钢轨表面局部缺陷生成领域，具体公开了一种面向钢轨表面局部缺陷场景的缺陷样本生成方法及系统，本发明方法以稳定扩散图像修复模型作为主干网络，通过低秩适配分支进行微调，并引入有效钢轨表面约束模块VRSC、类条件几何分解模块CGD和物理引导渲染模块PGR。本发明VRSC、CGD和PGR分别对应钢轨表面缺陷生成中的位置约束、几何约束和外观约束，本发明通过VRSC、CGD和PGR的顺序协同，按照由结构到形态、由形态到纹理的顺序共同约束扩散生成过程，使生成流程形成“先限定合理位置、再扩展缺陷形态、最后校正物理外观”的闭环，避免现有方法仅对扩散主干进行简单场景迁移。