A Yolov5 Model that detects business cards / ID cards / credit cards.

[this project] Identify Regions of Interest (ROI) containing the required information with deep learning [this project] Crop the regions identified above.

requirements:

• torch
• yolov5
• python3

pip install torch

git clone https://github.com/ultralytics/yolov5

move to yolov5 directory:

cd yolov5

then install requirements:

pip install -r requirements.txt

The trained Model id_card_detection.pt you can find it here in the Drive Link: Download it and move it to the folder /yolov5/runs/train/exp1/

python3 detect.py --source test_image.jpg --weights /yolov5/runs/train/exp1/id_card_detection.pt --save-crop

the argument save crop: will save the cropped image after detecting the boxes