The task was to implement the detection and recognition of car license plates within 2 hours
To run the project you need:
- Download weights from https://drive.google.com/drive/folders/1kz-_czvaMhZ0ubTsiPN2fOthPXyZzCGe?usp=sharing
- Put them in the folder ./weights
- run
pip install -r requirements.txt
- run
python main.py
with folowing arguments:--img_path
- for input image (e.g.-img_path /home/ivan/cars.jpeg
)--out_path
- for output json with OCR of license plates (e.g.--out_path temp_results
)
Plates detection is implemented with yolov5 model. OCR model was taken from https://github.com/sirius-ai/LPRNet_Pytorch and doesnt work properly at the time.
Results: