def main():
    base.mx_init() 
    
    modelpath = r'yolov8n-pose.om'
    DEVICE_ID = 0 # 设备id

    model = base.model(modelpath, deviceId=DEVICE_ID)
    image = cv2.imread('test.jpg')
    img = letterbox(image)
    data = pre_process(img)

    # print("预处理data为{}".format(data))
    print("数组尺寸为{}".format(data.shape))

    
    data = np.ascontiguousarray(data)
    input_tensor = Tensor(data)

   
    pred = model.infer([input_tensor])[0]

def pre_process(img):
    # 归一化 调整通道为（1，3，640，640）
    img = img / 255.
    img = np.transpose(img, (2, 0, 1))
    data = np.expand_dims(img, axis=0)
    return data