scottymcgee commited on
Commit
52abca3
·
verified ·
1 Parent(s): 6ee75af

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +54 -0
README.md ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Model Card for scottymcgee/image-classifier-stop-sign
2
+
3
+ This model classifies traffic-sign images as either **containing a stop sign** or **not containing a stop sign**.
4
+ It was trained with AutoGluon’s `MultiModalPredictor` on a binary image dataset of street signs.
5
+
6
+ ## Model Details
7
+
8
+ ### Model Description
9
+
10
+ - **Developed by:** Scotty McGee
11
+ - **Model type:** Image classifier (binary classification)
12
+ - **Languages (NLP):** Not applicable (vision model)
13
+ - **Finetuned from model:** Timm image backbone used by AutoGluon (default is EfficientNet or ResNet depending on config)
14
+
15
+ ### Model Sources
16
+
17
+ - **Repository:** https://huggingface.co/scottymcgee/image-classifier
18
+
19
+ ## Uses
20
+
21
+ ### Direct Use
22
+
23
+ Use this model to classify whether an input image contains a stop sign or not. It takes an RGB image as input and returns a predicted label and probabilities.
24
+
25
+ ### Downstream Use
26
+
27
+ It can be incorporated into larger perception systems (e.g., driver assistance, robotics) as a pre-screening classifier.
28
+
29
+ ### Out-of-Scope Use
30
+
31
+ Not intended for:
32
+ - Safety-critical deployment without further validation.
33
+ - Identifying other sign types beyond stop / no-stop.
34
+ - High-stakes enforcement or surveillance applications.
35
+
36
+ ## Bias, Risks, and Limitations
37
+
38
+ The model is trained on the specific dataset you provided. It may:
39
+ - Misclassify unusual or occluded stop signs.
40
+ - Perform poorly on non-U.S. stop sign shapes/colors if not present in training.
41
+ - Inherit any biases in the training images.
42
+
43
+ ### Recommendations
44
+
45
+ Always test on your target data before deployment. Combine with additional checks in safety-critical scenarios.
46
+
47
+ ## How to Get Started with the Model
48
+
49
+ ```python
50
+ from autogluon.multimodal import MultiModalPredictor
51
+
52
+ predictor = MultiModalPredictor.load("scottymcgee/image-classifier-stop-sign")
53
+ preds = predictor.predict(["example.jpg"])
54
+ print(preds)