高精度モデルを得るため

工業品質検査では、見逃し、過検出を極力抑さえる必要があります。そのため、欠陥セグメンテーションのモデル精度が非常に重要となります。本節では、最もモデルの品質に影響を与えるいくつかの要因および高品質なモデルをトレーニングする方法について紹介していきます。

ラベル付けの品質を確保する

ラベル付けの品質はモデルの精度に多大な影響を与えます。現場のプロジェクトにおいて、ラベル付けの品質が悪いせいでモデルの精度が出ないケースが90%以上と見られます。したがって、モデルの精度が出ない場合、最優先にラベル付けをチェックします。

ラベル付けの品質は、一貫性、完全性、精確性、確実性から考えます。

  1. 一貫性: 画像とカテゴリーのラベルは互いに対応することです。

    ../../../_images/label_consistency.png
  2. 完全性: 条件に満たすすべての領域に漏れなくラベルを付けることです。

    ../../../_images/label_completeness.png
  3. 精確性: ラベルの輪郭が対象の輪郭にピッタリ合わせることです。

    ../../../_images/label_accuracy.png
  4. 確実性: 人で判断できない疑似欠陥について、マスクツールを使用して欠陥を覆うことで回避することです。

    ../../../_images/certainty.png

注意

複数の疑似欠陥が含まれている画像は破棄します。

適切な関心領域(ROI)を設定する

ROIを設定することで、背景による干渉を抑えられます。ROIの境界は、なるべく対象物の外縁に近づかせます。

../../../_images/roi2.png

ヒント

ROI設定はすべての画像に適応されます。そのため、すべての画像にある対象物がROI内にあることを確認してください。特に、ワークの供給位置が不安定な場合や、サイズがバラバラな場合に注意しましょう。

適切なデータセットを選択する

  1. 画像枚数を控える

    欠陥セグメンテーションでは、初めてのモデル学習に20~30枚使用することを推奨します(欠陥の種類やバラツキによって決めてください)。データ量は多いほど良いわけではなく、あまりにも多いと、後でモデルを改善する時にトレーニング時間が延ばされます。

  2. データの多様性を確保する

    データセット画像は、検出する対象物の照明環境、色、サイズなどの情報を含んでいる必要があります。また、全種類のNG画像を含まなければなりません。OK画像同士の違いが少なく、特徴が似ている場合、適切に減らしても大丈夫です。

  3. 各種類のデータを均等な割合で使用する

    トレーニングセットに含まれる異なるカテゴリーの画像を均等な割合で使用する必要があります。あるカテゴリーの対象物の画像を20枚、別のカテゴリーの画像を3枚使用したりしてはいけません。

  4. トレーニングセットが使用シーンと一致する

    照明環境、ワークの特徴、検査時の背景、視野の広さなど、現場のシーンと一致する背景の画像を使用します。