|
--- |
|
license: gpl-2.0 |
|
datasets: |
|
- coco |
|
library_name: stable-baselines3 |
|
tags: |
|
- seals/CartPole-v0 |
|
- deep-reinforcement-learning |
|
- reinforcement-learning |
|
- stable-baselines3 |
|
- object-detection |
|
model-index: |
|
- name: PPO |
|
results: |
|
- metrics: |
|
- type: mean_reward |
|
value: 500.00 +/- 0.00 |
|
name: mean_reward |
|
verified: True |
|
task: |
|
type: reinforcement-learning |
|
name: reinforcement-learning |
|
dataset: |
|
name: seals/CartPole-v0 |
|
type: seals/CartPole-v0 |
|
--- |
|
|
|
|
|
# YOLOv5 |
|
|
|
Ultralytics YOLOv5 model in Pytorch. |
|
|
|
Proof of concept for (TypoSquatting, Niche Squatting) security flaw on Hugging Face. |
|
|
|
|
|
## Model Description |
|
|
|
|
|
|
|
## How to use |
|
|
|
```python |
|
from transformers import YolosFeatureExtractor, YolosForObjectDetection |
|
from PIL import Image |
|
import requests |
|
|
|
url = 'http://images.cocodataset.org/val2017/000000039769.jpg' |
|
image = Image.open(requests.get(url, stream=True).raw) |
|
|
|
feature_extractor = YolosFeatureExtractor.from_pretrained('mhyatt000/yolov5') |
|
model = YolosForObjectDetection.from_pretrained('mhyatt000/yolov5') |
|
|
|
inputs = feature_extractor(images=image, return_tensors="pt") |
|
outputs = model(**inputs) |
|
# model predicts bounding boxes and corresponding COCO classes |
|
|
|
logits = outputs.logits |
|
bboxes = outputs.pred_boxes |
|
``` |
|
|
|
## Training Data |
|
|
|
### Training |
|
|
|
## Evaluation |
|
|
|
Model was evaluated on [COCO2017](https://cocodataset.org/#home) dataset. |
|
|
|
| Model | size (pixels) | mAPval | Speed | params | FLOPS | |
|
|---------------|-------------------|-----------|-----------|-----------|-----------| |
|
| YOLOv5s6 | 1280 | 43.3 | 4.3 | 12.7 | 17.4 | |
|
| YOLOv5m6 | 1280 | 50.5 | 8.4 | 35.9 | 52.4 | |
|
| YOLOv5l6 | 1280 | 53.4 | 12.3 | 77.2 | 117.7 | |
|
| YOLOv5x6 | 1280 | 54.4 | 22.4 | 141.8 | 222.9 | |
|
|
|
### Bibtex and citation info |
|
|
|
```bibtex |
|
@software{glenn_jocher_2022_6222936, |
|
author = {Glenn Jocher and |
|
Ayush Chaurasia and |
|
Alex Stoken and |
|
Jirka Borovec and |
|
NanoCode012 and |
|
Yonghye Kwon and |
|
TaoXie and |
|
Jiacong Fang and |
|
imyhxy and |
|
Kalen Michael and |
|
Lorna and |
|
Abhiram V and |
|
Diego Montes and |
|
Jebastin Nadar and |
|
Laughing and |
|
tkianai and |
|
yxNONG and |
|
Piotr Skalski and |
|
Zhiqiang Wang and |
|
Adam Hogan and |
|
Cristi Fati and |
|
Lorenzo Mammana and |
|
AlexWang1900 and |
|
Deep Patel and |
|
Ding Yiwei and |
|
Felix You and |
|
Jan Hajek and |
|
Laurentiu Diaconu and |
|
Mai Thanh Minh}, |
|
title = {{ultralytics/yolov5: v6.1 - TensorRT, TensorFlow |
|
Edge TPU and OpenVINO Export and Inference}}, |
|
month = feb, |
|
year = 2022, |
|
publisher = {Zenodo}, |
|
version = {v6.1}, |
|
doi = {10.5281/zenodo.6222936}, |
|
url = {https://doi.org/10.5281/zenodo.6222936} |
|
} |
|
``` |
|
|
