Upload 5 files

README.md

@@ -1,3 +1,88 @@
----
-license: mit
----
+ 
+---
+tags:
+- yolov5
+- yolo
+- vision
+- object-detection
+- pytorch
+library_name: yolov5
+library_version: 7.0.6
+inference: false
+
+datasets:
+- keremberke/license-plate-object-detection
+
+model-index:
+- name: keremberke/yolov5n-license-plate
+  results:
+  - task:
+      type: object-detection
+
+    dataset:
+      type: keremberke/license-plate-object-detection
+      name: keremberke/license-plate-object-detection
+      split: validation
+
+    metrics:
+      - type: precision  # since mAP@0.5 is not available on hf.co/metrics
+        value: 0.9783431294995892  # min: 0.0 - max: 1.0
+        name: mAP@0.5
+---
+
+<div align="center">
+  <img width="640" alt="keremberke/yolov5n-license-plate" src="https://huggingface.co/keremberke/yolov5n-license-plate/resolve/main/sample_visuals.jpg">
+</div>
+
+### How to use
+
+- Install [yolov5](https://github.com/fcakyon/yolov5-pip):
+
+```bash
+pip install -U yolov5
+```
+
+- Load model and perform prediction:
+
+```python
+import yolov5
+
+# load model
+model = yolov5.load('keremberke/yolov5n-license-plate')
+  
+# set model parameters
+model.conf = 0.25  # NMS confidence threshold
+model.iou = 0.45  # NMS IoU threshold
+model.agnostic = False  # NMS class-agnostic
+model.multi_label = False  # NMS multiple labels per box
+model.max_det = 1000  # maximum number of detections per image
+
+# set image
+img = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
+
+# perform inference
+results = model(img, size=640)
+
+# inference with test time augmentation
+results = model(img, augment=True)
+
+# parse results
+predictions = results.pred[0]
+boxes = predictions[:, :4] # x1, y1, x2, y2
+scores = predictions[:, 4]
+categories = predictions[:, 5]
+
+# show detection bounding boxes on image
+results.show()
+
+# save results into "results/" folder
+results.save(save_dir='results/')
+```
+
+- Finetune the model on your custom dataset:
+
+```bash
+yolov5 train --data data.yaml --img 640 --batch 16 --weights keremberke/yolov5n-license-plate --epochs 10
+```
+
+**More models available at: [awesome-yolov5-models](https://github.com/keremberke/awesome-yolov5-models)**

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0e43f44b7354b2bc7de11ee8a3719fbb2de5bacd987f46e44a95380d6284240
+size 132

config.json

@@ -0,0 +1 @@
+{"input_size": 640, "task": "object-detection", "best_ap50": 0.9783431294995892}

events.out.tfevents.1672540597.ovyeman.11600.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:15af1e3c6663816b62cfbbe598b6eae84d31ff10b0c5a83971db9f460ff14cfb
+size 131