readme: ---
  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%**
  
  **fps**:
  - Nvidia Geforce GTX960, 4 GB: **69 fps**
  
  