---
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.

mAP with evaluation from pycocotools: 67%

mAP with evaluation from yolov5 val.py script: 71%