AuditEdge's picture
initial commit
81e13bb
import argparse
from asyncio.log import logger
from Layoutlm_inference.ocr import prepare_batch_for_inference
from Layoutlm_inference.inference_handler import handle
import logging
import os
if __name__ == "__main__":
# try:
parser = argparse.ArgumentParser()
parser.add_argument("--model_path", type=str)
parser.add_argument("--images_path", type=str)
args, _ = parser.parse_known_args()
images_path = args.images_path
image_files = os.listdir(images_path)
images_path = [images_path+f'/{image_file}' for image_file in image_files]
inference_batch = prepare_batch_for_inference(images_path)
context = {"model_dir": args.model_path}
output_ls = handle(inference_batch,context)
print("output_ls",output_ls)
# except Exception as err:
# os.makedirs('log', exist_ok=True)
# logging.basicConfig(filename='log/error_output.log', level=logging.ERROR,
# format='%(asctime)s %(levelname)s %(name)s %(message)s')
# logger = logging.getLogger(__name__)
# logger.error(err)