File size: 3,561 Bytes
3bc20d9 afc8888 |
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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 |
---
license: mit
metrics:
- accuracy
- precision
- recall
- confusion_matrix
tags:
- License-Plate-Character-Detector
- Character recognition
- ocr
- YOLOv8m
pipeline_tag: object-detection
---
# License Plate Character Detection Model
This repository contains a YOLOv8-based model for detecting characters in license plates. The model is trained to identify and localize individual characters on vehicle license plates, which can be useful for various applications such as automated parking systems, traffic monitoring, and vehicle identification.
## Model Details
- **Model Architecture**: YOLOv8
- **Task**: Character detection in license plates
- **Performance Metrics**: Accuracy, Precision, Recall
## Visual Demonstration
This image demonstrates the model's ability to detect and localize individual characters on a license plate. The bounding boxes show the detected characters.
## Installation
To use this model, you'll need to have Python installed along with the following dependencies:
```
pip install ultralytics
pip install torch
pip install huggingface_hub
```
## Usage
Here's a basic example of how to use the model:
```python
from ultralytics import YOLO
import cv2
import numpy as np
# Load the YOLOv8 model
model = YOLO('path/to/your/best.pt') # Load your trained model
# Read the image
image = cv2.imread('path/to/your/image.jpg')
# Run inference on the image
results = model(image)
# Process the results
for result in results:
boxes = result.boxes.cpu().numpy() # Get bounding boxes
for box in boxes:
# Get box coordinates
x1, y1, x2, y2 = box.xyxy[0]
x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
# Draw bounding box
cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
# If you have class names, you can add them to the image
if box.cls is not None:
label = f"{result.names[int(box.cls[0])]} {box.conf[0]:.2f}"
cv2.putText(image, label, (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
# Save or display the result
cv2.imwrite('output_image.jpg', image)
# Or to display (if running in an environment with GUI):
# cv2.imshow('Result', image)
# cv2.waitKey(0)
# cv2.destroyAllWindows()
```
## Training
If you want to train the model on your own dataset:
1. Prepare your dataset in the appropriate format for YOLOv8.
2. Use the YOLOv8 training script with your custom configuration.
## Model Performance
### Accuracy
Our model achieves an overall accuracy of [97.12]% on the test set. Here's a breakdown of accuracy for each character:
### Confusion Matrix
Below is the confusion matrix for our model, showing its performance across all characters:
This matrix provides insights into which characters are most often confused with each other, helping to identify areas for potential improvement.
### Additional Metrics
- Precision: [99.3]%
- Recall: [93.45]%
- mAP (mean Average Precision): [97.544]%
For any questions or feedback, please open an issue in this repository.
|