MSherbinii commited on
Commit
1fc697b
·
verified ·
1 Parent(s): 719a1cb

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +69 -0
README.md ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - anomaly-detection
5
+ - efficientad
6
+ - mvtec-ad
7
+ - pytorch
8
+ - computer-vision
9
+ datasets:
10
+ - mvtec_ad
11
+ metrics:
12
+ - auroc
13
+ ---
14
+
15
+ # EfficientAD - Bottle
16
+
17
+ Anomaly detection model trained on MVTec AD bottle dataset using EfficientAD.
18
+
19
+ ## Model Details
20
+
21
+ - **Architecture**: EfficientAD (medium)
22
+ - **Dataset**: MVTec AD - bottle
23
+ - **Task**: Anomaly Detection & Localization
24
+ - **Framework**: PyTorch
25
+
26
+ ## Files
27
+
28
+ - `teacher_final.pth`: Teacher network weights
29
+ - `student_final.pth`: Student network weights
30
+ - `autoencoder_final.pth`: Autoencoder network weights
31
+ - `normalization.pth`: Normalization parameters for inference
32
+
33
+ ## Usage
34
+
35
+ ```python
36
+ from huggingface_hub import hf_hub_download
37
+
38
+ # Download weights
39
+ teacher_path = hf_hub_download(
40
+ repo_id="MSherbinii/efficientad-bottle",
41
+ filename="teacher_final.pth"
42
+ )
43
+ student_path = hf_hub_download(
44
+ repo_id="MSherbinii/efficientad-bottle",
45
+ filename="student_final.pth"
46
+ )
47
+ autoencoder_path = hf_hub_download(
48
+ repo_id="MSherbinii/efficientad-bottle",
49
+ filename="autoencoder_final.pth"
50
+ )
51
+ normalization_path = hf_hub_download(
52
+ repo_id="MSherbinii/efficientad-bottle",
53
+ filename="normalization.pth"
54
+ )
55
+
56
+ # Load with PyTorch
57
+ import torch
58
+ teacher = torch.load(teacher_path, map_location='cpu')
59
+ student = torch.load(student_path, map_location='cpu')
60
+ autoencoder = torch.load(autoencoder_path, map_location='cpu')
61
+ ```
62
+
63
+ ## Citation
64
+
65
+ Based on EfficientAD: Accurate Visual Anomaly Detection at Millisecond-Level Latencies
66
+
67
+ ## License
68
+
69
+ MIT