arXiv reaDer
欠陥認識特徴操作による少数ショット欠陥画像生成
Few-Shot Defect Image Generation via Defect-Aware Feature Manipulation
欠陥検査のパフォーマンスは、業界で不十分な欠陥画像によって大幅に妨げられてきました。これは、データ拡張としてより多くのサンプルを生成することで軽減できます。我々は、困難な数ショットのケースにおける最初の欠陥画像生成方法を提案します。ほんの一握りの欠陥画像と比較的欠陥のない画像が与えられた場合、私たちの目標は、新しい欠陥画像でデータセットを増強することです。私たちの方法は、2 つのトレーニング段階で構成されています。まず、欠陥のない画像をバックボーンとして、データ効率の高い StyleGAN2 をトレーニングします。次に、欠陥認識残差ブロックをバックボーンにアタッチします。バックボーンは、合理的な欠陥マスクを生成することを学習し、それに応じて、追加されたモジュールを限られた欠陥画像でトレーニングすることにより、マスクされた領域内の機能を操作します。 MVTec AD データセットでの広範な実験は、現実的で多様な欠陥画像を生成する方法の有効性を検証するだけでなく、下流の欠陥検査タスクにもたらす利点も明らかにします。コードは https://github.com/Ldhlwh/DFMGAN で入手できます。
The performances of defect inspection have been severely hindered by insufficient defect images in industries, which can be alleviated by generating more samples as data augmentation. We propose the first defect image generation method in the challenging few-shot cases. Given just a handful of defect images and relatively more defect-free ones, our goal is to augment the dataset with new defect images. Our method consists of two training stages. First, we train a data-efficient StyleGAN2 on defect-free images as the backbone. Second, we attach defect-aware residual blocks to the backbone, which learn to produce reasonable defect masks and accordingly manipulate the features within the masked regions by training the added modules on limited defect images. Extensive experiments on MVTec AD dataset not only validate the effectiveness of our method in generating realistic and diverse defect images, but also manifest the benefits it brings to downstream defect inspection tasks. Codes are available at https://github.com/Ldhlwh/DFMGAN.
updated: Sat Mar 04 2023 11:43:08 GMT+0000 (UTC)
published: Sat Mar 04 2023 11:43:08 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト