therealestcoder commited on
Commit
8becd34
·
verified ·
1 Parent(s): e7991dc

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +54 -0
README.md ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - image-classification
5
+ - computer-vision
6
+ - defect-detection
7
+ - automotive
8
+ - pytorch
9
+ - timm
10
+ - efficientnet
11
+ language:
12
+ - ru
13
+ pipeline_tag: image-classification
14
+ ---
15
+
16
+ # Paint Defect Detector
17
+
18
+ A binary image classifier that detects **paint defects** on car body panels using transfer learning with EfficientNetV2-S backbone (via imm).
19
+
20
+ ## Model Architecture
21
+
22
+ - **Backbone**: EfficientNetV2-S (pretrained, from imm)
23
+ - **Head**: Dropout → Linear(feat_dim, 256) → GELU → Dropout → Linear(256, 2)
24
+ - **Task**: Binary classification — clean vs defect
25
+
26
+ ## Training
27
+
28
+ - **Optimizer**: AdamW with cosine annealing LR scheduler
29
+ - **Loss**: CrossEntropyLoss with label smoothing
30
+ - **Augmentations**: Albumentations pipeline
31
+ - **Metrics**: AUC-ROC, F1, Accuracy
32
+
33
+ ## Inference
34
+
35
+ The project includes a FastAPI REST API (src/api.py) for serving predictions, and a Grad-CAM visualisation layer for model explainability.
36
+
37
+ ## Project Structure
38
+
39
+ `
40
+ src/
41
+ config.py # Hyperparameters and paths
42
+ dataset.py # Dataset and data loaders
43
+ model.py # DefectClassifier model
44
+ train.py # Training loop
45
+ infer.py # Inference utilities
46
+ api.py # FastAPI inference server
47
+ prepare_data.py # Data preparation script
48
+ requirements.txt
49
+ `
50
+
51
+ ## Requirements
52
+
53
+ See
54
+ equirements.txt. Key dependencies: orch, imm, lbumentations, astapi, grad-cam.