언제나 그렇듯(!) 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 |
'프로그램 사용 > yolo_tensorflow' 카테고리의 다른 글
| moviad stfpm (0) | 2026.06.18 |
|---|---|
| ubuntu 26.04 + 3070 + tensorflow + python 3.14 + docker... (0) | 2026.06.17 |
| 딥러닝 학습 관련(epoch, loss) (0) | 2026.05.27 |
| NAS - Neural Architecture Search (0) | 2026.05.21 |
| MCUNet (0) | 2026.05.21 |
