STFPM 실행

언제나 그렇듯(!) 1080은 돌리기 힘든 시대구만.. 저번에 찾아둔 걸로 버전 맞추고

pip3 uninstall -y torch torchvision torchaudio xformers pip3 install torch==2.6.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 pip3 install xformers==0.0.29.post2

 

-mvtec-ad 는 dataset 다운로드 받아 압축푼 최상위 디렉토리를 지정해주면 된다.

~/src/STFPM/mvtec-ad$ tree -d -L 2 . ├── bottle │   ├── ground_truth │   ├── test │   └── train ├── cable │   ├── ground_truth │   ├── test │   └── train ├── capsule │   ├── ground_truth │   ├── test │   └── train ├── carpet │   ├── ground_truth │   ├── test │   └── train ├── grid │   ├── ground_truth │   ├── test │   └── train ├── hazelnut │   ├── ground_truth │   ├── test │   └── train ├── leather │   ├── ground_truth │   ├── test │   └── train ├── metal_nut │   ├── ground_truth │   ├── test │   └── train ├── pill │   ├── ground_truth │   ├── test │   └── train ├── screw │   ├── ground_truth │   ├── test │   └── train ├── tile │   ├── ground_truth │   ├── test │   └── train ├── toothbrush │   ├── ground_truth │   ├── test │   └── train ├── transistor │   ├── ground_truth │   ├── test │   └── train ├── wood │   ├── ground_truth │   ├── test │   └── train └── zipper     ├── ground_truth     ├── test     └── train

 

$ python3 main.py train --mvtec-ad  mvtec-ad --category carpet --epochs 200 ... [198/200] loss: 0.145916 [198/200] loss: 0.147569 [198/200] loss: 0.147157 [198/200] loss: 0.146806 [198/200] loss: 0.147341 [198/200] loss: 0.145462 [198/200] loss: 0.144424 Valid Loss: 0.0001142 [199/200] loss: 0.146722 [199/200] loss: 0.146246 [199/200] loss: 0.145853 [199/200] loss: 0.145966 [199/200] loss: 0.145793 [199/200] loss: 0.145863 [199/200] loss: 0.154857 Valid Loss: 0.0001460 real 9m35.951s user 31m24.301s sys 1m16.579s

 

loss 만 보고 학습을 하게 했네. 

그래도 200 epoch 인데 내부는 어떤지 모르겠지만 10분 컷이면 양호하네

 

cpu 로 1개 코어만 사용해서 하는데 제법 걸린다. 이미지로 판별하고 출력할 수 없나?

$ time python3 main.py test --mvtec-ad  mvtec-ad --category carpet --checkpoint snapshots/carpet/best.pth.tar load snapshots/carpet/best.pth.tar Catergory: carpet Pixel-AUC: 0.990288 Image-AUC: 0.955859 PRO: 0.959637 real 5m39.944s user 5m46.592s sys 0m1.064s

 

tar 라면서 왜 zip 이래?!?

~/src/STFPM/snapshots/carpet$ file * best.pth.tar: Zip archive data, at least v0.0 to extract, compression method=store ~/src/STFPM/snapshots/carpet$ ll -h total 43M drwxrwxr-x 2 falinux falinux 4.0K  6월  1 11:57 ./ drwxrwxr-x 3 falinux falinux 4.0K  6월  1 11:14 ../ -rw-rw-r-- 1 falinux falinux  43M  6월  1 11:52 best.pth.tar

 

[링크 : https://github.com/gdwang08/STFPM]