metadata
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%