PyTorch随笔 - 获取TensorRT(TRT)模型输入和输出

获取TensorRT(TRT)模型输入和输出，用于创建TRT的模型服务使用，具体参考脚本check_trt_script.py，如下：

• 脚本输入：TRT的模型路径和输入图像尺寸
• 脚本输出：模型的输入和输出结点信息，同时验证TRT模型是否可用

#!/usr/bin/env python
# -- coding: utf-8 --
"""
Copyright (c) 2021. All rights reserved.
Created by C. L. Wang on 16.9.21
"""

import argparse

import numpy as np


def check_trt(model_path, image_size)<