Update README.md

README.md

@@ -37,28 +37,28 @@ model-index:
     metrics:
     - type: precision
       name: Precision
-      value: 0.9078
+      value: 0.844
     - type: recall
       name: Recall
-      value: 0.9064
+      value: 0.847
     - type: mAP50
       name: mAP50
-      value: 0.9589
+      value: 0.893
     - type: mAP50-95
       name: mAP50-95
-      value: 0.6088
+      value: 0.546
 ---
 
 # Model for detecting Hardhats and Hats
 
 
 <div align="center">
-  <img width="640" alt="luisarizmendi/safety-equipment" src="https://huggingface.co/luisarizmendi/hardhat-or-hat/resolve/main/example.png">
+  <img width="640" alt="luisarizmendi/hardhat-or-hat" src="example.png">
 </div>
 
 ## Model binary
 
-You can [download the model from here](https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detector-hardhat-or-hat/object-detector-hardhat-or-hat.pt)
+You can [download the model from here](https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detection-hardhat-or-hat/object-detection-hardhat-or-hat-m.pt)
 
 
 ## Labels
@@ -70,11 +70,24 @@ You can [download the model from here](https://github.com/luisarizmendi/ai-apps/
 ```
 
 
+## Base Model
+
+Ultralytics/YOLO11m
+
 ## Model metrics
 
+```
+YOLO11m summary (fused): 303 layers, 20,032,345 parameters, 0 gradients, 67.7 GFLOPs
+                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95)
+                   all       1992      15306      0.844      0.847      0.893      0.546
+                   hat        244        287      0.869      0.811      0.876      0.578
+                helmet       1202       3942      0.916      0.892      0.942       0.61
+             no_helmet        741      11077      0.746      0.838      0.861       0.45
+```
+
 
 <div align="center">
-  <img width="640" alt="luisarizmendi/safety-equipment" src="https://huggingface.co/luisarizmendi/hardhat-or-hat/resolve/main/confusion_matrix_normalized.png"> <img width="640" alt="luisarizmendi/safety-equipment" src="https://huggingface.co/luisarizmendi/hardhat-or-hat/resolve/main/results.png">
+  <img width="640" alt="luisarizmendi/hardhat-or-hat" src="confusion_matrix_normalized.png"> <img width="640" alt="luisarizmendi/hardhat-or-hat" src="results.png">
 </div>
 
 
@@ -83,12 +96,9 @@ You can [download the model from here](https://github.com/luisarizmendi/ai-apps/
 [https://universe.roboflow.com/luisarizmendi/hardhat-or-hat](https://universe.roboflow.com/luisarizmendi/hardhat-or-hat)
 
 
-
 ## Model training
 
-### Notebook
-
-You can [review the Jupyter notebook here](https://huggingface.co/luisarizmendi/hardhat-or-hat/blob/main/train.ipynb)
+You can [review the Jupyter notebook here](https://github.com/luisarizmendi/ai-apps/blob/main/dev/hardhat-or-hat/train.ipynb)
 
 ### Hyperparameters
 
@@ -125,8 +135,7 @@ mixup=0.0,   # Image mixup
 ```
 
 
-## Usage
-
+## Model Usage
 
 ### Usage with Huggingface spaces
 
@@ -139,6 +148,7 @@ Remember to check that the Model URL is pointing to the model that you want to t
 </div>
 
 
+
 ### Usage with Python script
 
 Install the following PIP requirements
@@ -151,7 +161,8 @@ opencv-python
 torch
 ```
 
-Then [run the python code below](https://huggingface.co/luisarizmendi/hardhat-or-hat/blob/main/run_model.py) and then open `http://localhost:7860` in a browser to upload and scan the images.
+Then [run the python code below ](https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detector-hardhat-or-hat/run_model.py) and open `http://localhost:7860` in a browser to upload and scan the images.
+
 
 ```
 import gradio as gr
@@ -161,7 +172,7 @@ import os
 import cv2
 import torch
 
-DEFAULT_MODEL_URL = "https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detector-hardhat-or-hat/object-detector-hardhat-or-hat.pt"
+DEFAULT_MODEL_URL = "https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detection-hardhat-or-hat/object-detection-hardhat-or-hat-m.pt"
 
 def detect_objects_in_files(model_input, files):
     """
@@ -213,4 +224,9 @@ interface = gr.Interface(
 
 if __name__ == "__main__":
     interface.launch()
-```
+
+```
+
+
+
+