Commit
•
1c85630
1
Parent(s):
09deda7
minor readme adjustments
Browse files
README.md
CHANGED
|
@@ -85,6 +85,8 @@ A live demonstration is freely accesible [here](https://croppie.org/).
|
|
| 85 |
|
| 86 |
The model has been trained using the custom YOLO class found in ```./scripts/custom_YOLO.py```. The custom YOLO class can be exactly used as the original [YOLO class](https://docs.ultralytics.com/reference/models/yolo/model/). The hyperparameters used during the training can be found in ```./scripts/args.yaml```.
|
| 87 |
|
|
|
|
|
|
|
| 88 |
## Test metrics
|
| 89 |
|
| 90 |
<img src="images/F1_curve.png" width="300">
|
|
|
|
| 85 |
|
| 86 |
The model has been trained using the custom YOLO class found in ```./scripts/custom_YOLO.py```. The custom YOLO class can be exactly used as the original [YOLO class](https://docs.ultralytics.com/reference/models/yolo/model/). The hyperparameters used during the training can be found in ```./scripts/args.yaml```.
|
| 87 |
|
| 88 |
+
The training maximize the [email protected], which is the mean Average Precision calculated at a 0.5 Intersection over Union (IoU) threshold, measuring how well the model detects objects with at least 50% overlap between predicted and ground truth bounding boxes.
|
| 89 |
+
|
| 90 |
## Test metrics
|
| 91 |
|
| 92 |
<img src="images/F1_curve.png" width="300">
|