jonalvarez01 commited on
Commit
ab2ec47
·
verified ·
1 Parent(s): 9e6f8fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -1
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" />