arXiv reaDer
自動表面欠陥セグメントのニューラルスタイル転送による欠陥サンプルシミュレーション
Defective samples simulation through Neural Style Transfer for automatic surface defect segment
  工業製品の品質検査には欠陥サンプルがないため、トレーニングされたセグメンテーションモデルはオンラインで適用すると過剰適合する傾向があります。この問題に対処するために、ニューラルスタイル転送に基づく欠陥サンプルシミュレーションアルゴリズムを提案します。シミュレーションアルゴリズムは、トレーニングに少数の欠陥サンプルのみを必要とし、次のステップのセグメンテーションタスクのシミュレーションサンプルを効率的に生成できます。私たちの作業では、マスクされたヒストグラムマッチングモジュールを導入して、生成された領域と真の欠陥の色の一貫性を維持します。周囲のピクセルとのテクスチャの一貫性を維持するために、高速スタイル転送アルゴリズムを使用して、生成された領域を背景にブレンドします。同時に、ヒストグラム損失も使用して、生成される画像の品質をさらに向上させます。また、セグメントネットの新しい構造を提案し、欠陥セグメンテーションタスクにより適したものにします。ボタンデータセット上で、実際の欠陥サンプルと生成されたシミュレーションサンプルでセグメントネットを個別にトレーニングします。結果は、生成されたシミュレーションサンプルのみでトレーニングされたモデルのF1スコアが0.80に達し、実際のサンプル結果よりも優れていることを示しています。
Owing to the lack of defect samples in industrial product quality inspection, trained segmentation model tends to overfit when applied online. To address this problem, we propose a defect sample simulation algorithm based on neural style transfer. The simulation algorithm requires only a small number of defect samples for training, and can efficiently generate simulation samples for next-step segmentation task. In our work, we introduce a masked histogram matching module to maintain color consistency of the generated area and the true defect. To preserve the texture consistency with the surrounding pixels, we take the fast style transfer algorithm to blend the generated area into the background. At the same time, we also use the histogram loss to further improve the quality of the generated image. Besides, we propose a novel structure of segment net to make it more suitable for defect segmentation task. We train the segment net with the real defect samples and the generated simulation samples separately on the button datasets. The results show that the F1 score of the model trained with only the generated simulation samples reaches 0.80, which is better than the real sample result.
updated: Tue Oct 08 2019 10:55:19 GMT+0000 (UTC)
published: Tue Oct 08 2019 10:55:19 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アソシエイト