Erpix3lt commited on
Commit
bc280ea
·
verified ·
1 Parent(s): 2f0798f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +68 -0
README.md CHANGED
@@ -1,3 +1,71 @@
1
  ---
2
  license: mit
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ language:
4
+ - en
5
  ---
6
+
7
+ ---
8
+ language: en
9
+ tags:
10
+ - computer-vision
11
+ - image-classification
12
+ - traffic-signs
13
+ - pytorch
14
+ license: mit
15
+ datasets:
16
+ - https://platform.ultralytics.com/maaaaaaaaaaaaaaaax/datasets/gtsrb-full
17
+ pipeline_tag: image-classification
18
+ ---
19
+
20
+ # Traffic Sign Classifier (GTSRB)
21
+
22
+ ## Model Summary
23
+ This is a traffic sign classification model trained on GTSRB classes.
24
+ Architecture: lightweight CNN with feature extractor, 256-d bottleneck, and linear classifier.
25
+
26
+ This checkpoint is intended for research and analysis workflows, not safety-critical deployment.
27
+
28
+ ## Dataset
29
+ - Source: https://platform.ultralytics.com/maaaaaaaaaaaaaaaax/datasets/gtsrb-full
30
+ - Domain: German traffic sign recognition
31
+ - Image preprocessing:
32
+ - Resize to 48 x 48
33
+ - RGB input
34
+ - Training augmentation:
35
+ - Random rotation: ±10 degrees
36
+ - Color jitter: brightness 0.2, contrast 0.2
37
+
38
+ ## Model Architecture
39
+ - Backbone: custom CNN
40
+ - Block 1:
41
+ - Conv(3,32), BN, ELU
42
+ - Conv(32,32), BN, ELU
43
+ - MaxPool2d(2), Dropout2d(0.2)
44
+ - Block 2:
45
+ - Conv(32,64), BN, ELU
46
+ - Conv(64,64), BN, ELU
47
+ - MaxPool2d(2), Dropout2d(0.3)
48
+ - Block 3:
49
+ - Conv(64,128), BN, ELU
50
+ - Conv(128,128), BN, ELU
51
+ - MaxPool2d(2), Dropout2d(0.4)
52
+ - Bottleneck:
53
+ - Flatten
54
+ - Linear(128 x 6 x 6 -> 256), ELU, Dropout(0.5)
55
+ - Head:
56
+ - Linear(256 -> 42)
57
+
58
+ ## Framework and Weights
59
+ - Framework: PyTorch
60
+ - Weight format: state_dict checkpoint
61
+
62
+ ## Intended Uses
63
+ - Research on traffic sign recognition
64
+ - Transfer learning experiments
65
+ - Educational use for compact CNN pipelines
66
+
67
+ ## Out-of-Scope Uses
68
+ - Real-world safety-critical decision making
69
+
70
+ ## Limitations
71
+ - Trained on a narrow visual domain (GTSRB)