File size: 553 Bytes
7b434b7 47c0f7d 2f74d3e 7b434b7 9a41436 8ea4338 61fb4b9 9a41436 74be20e 61fb4b9 74be20e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
---
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**
|