Update README.md

README.md

@@ -3,29 +3,32 @@ license: apache-2.0
 base_model: hustvl/yolos-small
 tags:
 - generated_from_trainer
+- medical
 model-index:
 - name: yolos-small-Axial_MRIs
   results: []
+datasets:
+- Francesco/axial-mri
+language:
+- en
+pipeline_tag: object-detection
 ---
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
-
 # yolos-small-Axial_MRIs
 
-This model is a fine-tuned version of [hustvl/yolos-small](https://huggingface.co/hustvl/yolos-small) on an unknown dataset.
+This model is a fine-tuned version of [hustvl/yolos-small](https://huggingface.co/hustvl/yolos-small).
 
 ## Model description
 
-More information needed
+For more information on how it was created, check out the following link: https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/blob/main/Computer%20Vision/Object%20Detection/Axial%20MRIs/Axial_MRIs_Object_Detection_YOLOS.ipynb
 
 ## Intended uses & limitations
 
-More information needed
+This model is intended to demonstrate my ability to solve a complex problem using technology.
 
 ## Training and evaluation data
 
-More information needed
+Dataset Source: https://huggingface.co/datasets/Francesco/axial-mri
 
 ## Training procedure
 
@@ -42,11 +45,24 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-
+| Metric Name | IoU | Area| maxDets | Metric Value |
+|:-----:|:-----:|:-----:|:-----:|:-----:|
+| Average Precision (AP) | IoU=0.50:0.95 | all | maxDets=100 | 0.284 |
+| Average Precision (AP) | IoU=0.50 | all | maxDets=100 | 0.451 |
+| Average Precision (AP) | IoU=0.75 | all | maxDets=100 | 0.351 |
+| Average Precision (AP) | IoU=0.50:0.95 | small | maxDets=100 | 0.000 |
+| Average Precision (AP) | IoU=0.50:0.95 | medium | maxDets=100 | 0.182 |
+| Average Precision (AP) | IoU=0.50:0.95 | large | maxDets=100 | 0.663 |
+| Average Recall (AR) | IoU=0.50:0.95 | all | maxDets=1 | 0.388 |
+| Average Recall (AR) | IoU=0.50:0.95 | all | maxDets=10 | 0.524 |
+| Average Recall (AR) | IoU=0.50:0.95 | all | maxDets=100 | 0.566 |
+| Average Recall (AR) | IoU=0.50:0.95 | small | maxDets=100 | 0.000 |
+| Average Recall (AR) | IoU=0.50:0.95 | medium | maxDets=100 | 0.502 |
+| Average Recall (AR) | IoU=0.50:0.95 | large | maxDets=100 | 0.791 |
 
 ### Framework versions
 
 - Transformers 4.31.0
 - Pytorch 2.0.1+cu118
 - Datasets 2.14.3
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3