import argparse

from paddleocr import LayoutDetection





def parse_args():

    parser = argparse.ArgumentParser()

    parser.add_argument("--image_dir", type=str, default="layout.jpg")

    parser.add_argument("--model_name", type=str, default="PP-DocLayoutV3")

    parser.add_argument("--model_dir", type=str, default="PP-DocLayoutV3-weight")

    return parser.parse_args()





if __name__ == "__main__":

    args = parse_args()



    model = LayoutDetection(

        model_name=args.model_name,

        model_dir=args.model_dir,

        device="npu"

    )



    output = model.predict(args.image_dir, threshold=0.3, filter_overlap_boxes=False)

    for res in output:

        res.print()

        res.save_to_img("output")

        res.save_to_json("output")