| language: | |
| - en | |
| tags: | |
| - pytorch | |
| - yolov5 | |
| datasets: | |
| - pwmfd | |
| metrics: | |
| - coco | |
| Optimized YOLOv5 model trained on the PWMFD medical masks dataset using transfer learning from COCO with frozen backbone, data augmentations such as mosaic, and an input image size of 320 x 320. | |
| **Architecture:** [here](https://huggingface.co/joangog/pwmfd-yolov5/tensorboard?scroll=1#graphs&run=.) | |
| **AP:** | |
| - Evaluation from pycocotools: 67% | |
| - Evaluation from yolov5 val.py script: 71% | |