python3 onnx 推理Demo

最新推荐文章于 2025-10-02 14:56:00 发布

原创

最新推荐文章于 2025-10-02 14:56:00 发布 · 2k 阅读

5 ·

CC 4.0 BY-SA版权

博客提供了人脸68个关键点推理演示代码，聚焦于信息技术领域中图像处理相关内容，可用于人脸标定等应用。

1. 人脸68个关键点推理演示代码

#! /bin/python
# -*- coding: utf-8 -*-

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import warnings

import cv2
import onnx
import torch
import numpy as np
import onnxruntime

warnings.filterwarnings("ignore")


class ONNXModel(object):
    def __init__(self, onnx_path):
        """
        :param onnx_path:
        """
        self.onnx_session = onnxruntime.InferenceSession(onnx_path)
        self.input_name = self.get_input_name(self.onnx_session)
        self.output_name = self.get_output_name(self.onnx_session)
        print("input_name:{}".format(self.input_name))
        print("output_name:{}".format(self.output_name))

    def get_output_name(self, onnx_session):
        """
        output_name = onnx_session.get_outputs()[0].name
        :param onnx_session:
        :return:
        """
        output_name = []
        for node in onnx_session.get