MareArts Computer Vision Study.: 2024.11

11/03/2024

# install

pip install marearts-anpr

# code

# pip install marearts-anpr

import cv2

from PIL import Image

from marearts_anpr import ma_anpr_detector

from marearts_anpr import ma_anpr_ocr

from marearts_anpr import marearts_anpr_from_pil

from marearts_anpr import marearts_anpr_from_image_file

from marearts_anpr import marearts_anpr_from_cv2

if __name__ == '__main__':

    #################################

    ## Initiate MareArts ANPR

    print("EU ANPR")

    user_name = "your_email"

    serial_key = "your_serial_key"

    detector_model_version = "middle" # Options: refer to detector model table

    ocr_model_version = "eu" # Options: refer to ocr model table

    # MareArts ANPR Detector Inference

    anpr_d = ma_anpr_detector(detector_model_version, user_name, serial_key, conf_thres=0.3, iou_thres=0.5)

    # MareArts ANPR OCR Inference

    anpr_r = ma_anpr_ocr(ocr_model_version, user_name, serial_key)

    #################################

    #################################

    # Routine Task 1 - Predict from File

    image_path = './sample_images/eu_test1.jpg'

    output = marearts_anpr_from_image_file(anpr_d, anpr_r, image_path)

    print(output)

    # Routine Task 2 - Predict from cv2

    img = cv2.imread(image_path)

    output = marearts_anpr_from_cv2(anpr_d, anpr_r, img)

    print(output)

    # Routine Task 3 - Predict from Pillow

    pil_img = Image.open(image_path)

    output = marearts_anpr_from_pil(anpr_d, anpr_r, pil_img)

    print(output)

    #################################

    #################################

    ## Initiate MareArts ANPR for Korea

    print("ANPR Korean")

    # user_name, serial_key are already defined

    # anpr_d is also already initiated before

    ocr_model_version = "kr"

    # MareArts ANPR OCR Inference

    anpr_r = ma_anpr_ocr(ocr_model_version, user_name, serial_key)

    #################################

    # Routine Task 1 - Predict from File

    image_path = './sample_images/kr_test2.jpg'

    output = marearts_anpr_from_image_file(anpr_d, anpr_r, image_path)

    print(output)

    # Routine Task 2 - Predict from cv2 

    img = cv2.imread(image_path)

    output = marearts_anpr_from_cv2(anpr_d, anpr_r, img)

    print(output)

    # Routine Task 3 - Predict from Pillow

    pil_img = Image.open(image_path)

    output = marearts_anpr_from_pil(anpr_d, anpr_r, pil_img)

    print(output)

    #################################