ONNX PaddleOCR inference code

test.py

import os
import cv2
import numpy as np
from util import detectionclass as net

detection = net.Detection('./weights/detection.onnx')

def main():
    frame = cv2.imread('./images/plate.jpg')
    image = frame.copy()
    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

    points = detection(frame)
    points = net.sort_polygon(list(points))

    # draw detected polygon
    for point in points:
        point = np.array(point, dtype=np.int32)
        cv2.polylines(image,
                      [point], True,
                      (0, 255, 0), 2)

    cv2.imshow('Detection',image)
    cv2.waitKey(0)
    
if __name__ == '__main__':
    main()