A package for reading id and name on KTP and SIM
Project description
Indo OCR Army
This is a package for OCR Identity card in Indonesia.
for comprehensive use, you can do this:
import os
import cv2
import matplotlib.pyplot as plt
from IndoOCRArmy.modelOCR import numericalDetectron2, boundingBoxesDetectron2, alphabeticalDetectron2, easypredict
from IndoOCRArmy.drawer import DrawOCR
from config import cfg
# load classes
drawer_ocr = DrawOCR(cfg['drawOCR'])
bBoxDet = boundingBoxesDetectron2(cfg['boundingBoxesDetectron2'])
numDet = numericalDetectron2(cfg['numericalDetectron2'])
alphaDet = alphabeticalDetectron2(cfg['alphabeticalDetectron2'])
# load image
image_ktp = cv2.imread("assets/ktp_example.jpg")
image_sim = cv2.imread("assets/sim_example.jpg")
# detect boundingboxes
crops, boxes, labels = bBoxDet.predict(image_ktp, input_type='ktp')
# detect number and alphabet
dict_ID = numDet.predict_ensemble(crops[0])
dict_Name = alphaDet.predict_ensemble(crops[1])
# choose `weighted_hardvote_word` for the best result according to our benchmark
ID = dict_ID.get("weighted_hardvote_word")
Name = dict_Name.get("weighted_hardvote_word")
# parse NIK to get information about : location, gender, and birthdate
parse_NIK = numDet.parse_nik(ID)
# create listdata and listlabel for visualization later
listdata = [ID, Name]
listlabel = [x for x in list(labels.values()) if x is not None]
for label, data in parse_NIK.items():
listdata.append(data)
listlabel.append(label)
print(ID)
print(Name)
drawer_ocr.show_list_images(list_img=crops.values())
For comprehensive result, visualize this:
drawer_ocr.show_desc(image_ktp, boxes, labels, listdata, listlabel)
For quick result, use this:
from IndoOCRArmy.modelOCR import numericalDetectron2, boundingBoxesDetectron2, alphabeticalDetectron2, easypredict
import cv2
# load classes
drawer_ocr = DrawOCR(cfg['drawOCR'])
bBoxDet = boundingBoxesDetectron2(cfg['boundingBoxesDetectron2'])
numDet = numericalDetectron2(cfg['numericalDetectron2'])
alphaDet = alphabeticalDetectron2(cfg['alphabeticalDetectron2'])
image_ktp = cv2.imread("assets/ktp_example.jpg")
easypredict(image_ktp, bBoxDet, numDet, alphaDet, input_type='ktp')
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
indOCRArmy-0.1.7.tar.gz
(11.0 kB
view hashes)
Built Distribution
indOCRArmy-0.1.7-py3-none-any.whl
(11.8 kB
view hashes)
Close
Hashes for indOCRArmy-0.1.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 58aa802733c905677746d1c6eb17161cd6aa9eff4cb09a2a0d2844d3bb67dd99 |
|
MD5 | 495c5477d809f82ff4369f9887eb61d4 |
|
BLAKE2b-256 | fa3635a027fc22bd0fb19c523725f9919f16fe788c7bd975f0f9a60eaf7eca3d |