language: | |
- en | |
tags: | |
- pytorch | |
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. |