search

MiniBullLab / easy_ai

3 stars 1 forks source link

异常检测调研 #108

Closed lpj0822 closed 2 years ago

lpj0822 commented 3 years ago

https://blog.csdn.net/qq_36560894/article/details/110727751

kingwangxiang commented 3 years ago

1.使用Mvtec数据集中的pill数据测试杜伦大学提出GANomaly。 1.1实验迭代100epoch，实验图片尺寸为128X128，如图1-A所示(样本原图为800X800)，模型中间迭代结果如图1-B所示。 000 图1-Apill样本原图 pill_experience 图1-B模型迭代中间结果 1.2auc评价在0.6左右徘徊，部分结果如图2-A所示。 pill_auc 图2-A模型评估指标

lpj0822 commented 3 years ago

https://blog.csdn.net/weixin_38328533/article/details/114596655 @kingwangxiang 看看这个方法，有没有价值

foww-0001 commented 3 years ago

可以看一下这篇论文：CutPaste_Self-Supervised Learning for Anomaly Detection and Localization https://blog.csdn.net/weixin_43540533/article/details/116159775

foww-0001 commented 3 years ago

PaDiM code：https://github.com/taikiinoue45/PaDiM

foww-0001 commented 3 years ago

PatchCore anomaly detection: https://github.com/hcw-00/PatchCore_anomaly_detection https://github.com/taikiinoue45/PatchCore.git

foww-0001 commented 3 years ago

Student-Teacher Feature Pyramid Matching for Unsupervised Anomaly Detection implementation: https://github.com/hcw-00/STPM_anomaly_detection

foww-0001 commented 3 years ago

异常检测调研文档：百度网盘/内部文档/学习资料/基础调研/

kingwangxiang commented 3 years ago

1.尝试调通PaDiM code：https://github.com/taikiinoue45/PaDiM 1.1有两个库hydra和mlflow不知道如何安装。

foww-0001 commented 3 years ago

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

foww-0001 commented 3 years ago

PatchCore保存MemoryBank:

python3 easyai/tools/feature_save/one_class_feature_save.py -t one_class -i /home/wfw/data/VOCdevkit/MVTec/bottle/ImageSets/train.txt -m PatchCoreNet -w /home/wfw/workspace/HASCO/all_wights/wide_resnet50_2_patchcore.pt -c example/patchcore_config.json

PatchCore运行test:

python3 easyai/test_task.py -t one_class -i /home/wfw/data/VOCdevkit/MVTec/bottle/ImageSets/val.txt -m PatchCoreNet -w  /home/wfw/workspace/HASCO/all_wights/wide_resnet50_2_patchcore.pt -c example/patchcore_config.json
foww-0001 commented 3 years ago

模型上传到百度网盘：wide_resnet50_2_patchcore.pt

foww-0001 commented 3 years ago

Padim保存mean和covariance:

python3 easyai/tools/feature_save/one_class_feature_save.py -t one_class -i /home/wfw/data/VOCdevkit/MVTec/bottle/ImageSets/train.txt -m PadimNet -w /home/wfw/workspace/HASCO/all_wights/wide_resnet50_2_patchcore.pt -c example/padim_config.json

Padim运行test:

python3 easyai/test_task.py -t one_class -i /home/wfw/data/VOCdevkit/MVTec/bottle/ImageSets/val.txt -m PadimNet -w  /home/wfw/workspace/HASCO/all_wights/wide_resnet50_2_patchcore.pt -c example/padim_config.json
foww-0001 commented 3 years ago 
./easy_tools/train_scripts/OneClass.sh /home/wfw/data/VOCdevkit/MVTec/bottle/ImageSets/train.txt /home/wfw/data/VOCdevkit/MVTec/bottle/ImageSets/val.txt
kingwangxiang commented 3 years ago

上下位机，可以运行测试。需要config文件，后续要再调整下。

lpj0822 commented 3 years ago

将ganormaly配置好 @foww-0001

lpj0822 commented 3 years ago

把脚本也加进去

foww-0001 commented 3 years ago

异常检测轴承数据已添加，模型densenet161模型已上传

foww-0001 commented 3 years ago

data已经上传，用以下链接下载解压放到easy_tools下面，

http://118.31.19.101:8080/easy_tools/data_v3.zip

运行以下脚步进行训练：

./easy_tools/train_scripts/OneClass.sh /home/wfw/data/VOCdevkit/MVTec/bottle/ImageSets/train.txt /home/wfw/data/VOCdevkit/MVTec/bottle/ImageSets/val.txt
foww-0001 commented 3 years ago

测试新模型data可以正常运行得到结果。

foww-0001 commented 2 years ago

项目暂时关闭，后续有新的需求再新建。