search

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK
2.04k stars 323 forks source link

ADD PaDiM-Anomaly-Detection-Localization-master #401

Closed kyakuno closed 3 years ago

kyakuno commented 3 years ago

https://github.com/xiahaifeng1995/PaDiM-Anomaly-Detection-Localization-master Apache

kyakuno commented 3 years ago

最も性能が高い https://paperswithcode.com/sota/anomaly-detection-on-mvtec-ad

kyakuno commented 3 years ago

ailia-modelsのカテゴリはanomaly_detectionを新規追加。

kyakuno commented 3 years ago

resnet18のpretrained modelの各レイヤーのfeature mapを使ってvectorを作り、分散を計算、そこから離れたものを以上とする。

kyakuno commented 3 years ago

再学習不要の異常検知なのに、最も性能が高くなっている。

kyakuno commented 3 years ago

ailiaの場合はget_blob_dataを使って中間データを取得可能。

kyakuno commented 3 years ago

機能1：データセットフォルダを指定してそこから分散を計算してファイルに保存 機能２：画像ファイルを指定してセグメンテーションと正常/異常を判定する

ooe1123 commented 3 years ago

torch1.7、opset_version=11でエクスポートしたとき、それぞれのモデルの各レイヤのアウトプット名は次のようになった。

○ wide_resnet50_2 [layer1] 355 -> Relu_24 -> 356 (n,256,56,56) [layer2] 397 -> Relu_53 -> 398 (n,512,28,28) [layer3] 459 -> Relu_96 -> 460 (n,1024,14,14)

○ resnet18 [layer1] 139 -> Relu_12 -> 140 (n,64,56,56) [layer2] 155 -> Relu_23 -> 156 (n,128,28,28) [layer3] 171 -> Relu_34 -> 172 (n,256,14,14)