Update README.md
Browse files
README.md
CHANGED
@@ -51,4 +51,25 @@ results = model.train(data="data.yaml", epochs=100)
|
|
51 |
|
52 |
## Dataset
|
53 |
|
54 |
-
The model was trained on a custom dataset containing images of mobile phones and caps. The dataset was structured following YOLO format requirements.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
|
52 |
## Dataset
|
53 |
|
54 |
+
The model was trained on a custom dataset containing images of mobile phones and caps. The dataset was structured following YOLO format requirements.
|
55 |
+
|
56 |
+
|
57 |
+
## Model Performance Metrics
|
58 |
+
|
59 |
+
The model's performance was evaluated over 100 epochs of training. Here are the key metrics:
|
60 |
+
|
61 |
+
<img src="metrics/results.png" width="600" />
|
62 |
+
|
63 |
+
### Confusion Matrix
|
64 |
+
|
65 |
+
<img src="metrics/confusion_matrix.png" width="400" />
|
66 |
+
|
67 |
+
### Precision-Recall Curve
|
68 |
+
|
69 |
+
|
70 |
+
<img src="metrics/P_curve.png" width="400" />
|
71 |
+
<img src="metrics/R_curve.png" width="400" />
|
72 |
+
|
73 |
+
### F1-Score Curve
|
74 |
+
|
75 |
+
<img src="metrics/F1_curve.png" width="400" />
|