|
--- |
|
language: |
|
- en |
|
tags: |
|
- pytorch |
|
- yolov5 |
|
- object-detection |
|
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** |
|
|
|
|
