egorovivannn / dummy_license_plate

Project for license plate detection and recognition TEST TASK.

The task was to implement the detection and recognition of car license plates within 2 hours

To run the project you need:

1. Download weights from https://drive.google.com/drive/folders/1kz-_czvaMhZ0ubTsiPN2fOthPXyZzCGe?usp=sharing
2. Put them in the folder ./weights
3. run pip install -r requirements.txt
4. 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: