yolov3-tiny-onnx-TensorRT.zip

tensorrt转换，本人代码，盗版必究，禁止商用，

#
# Copyright 1993-2018 NVIDIA Corporation.  All rights reserved.
#
# NOTICE TO LICENSEE:
#
# This source code and/or documentation （“Licensed Deliverables“） are
# subject to NVIDIA intellectual property rights under U.S. and
# international Copyright laws.
#
# These Licensed Deliverables contained herein is PROPRIETARY and
# CONFIDENTIAL to NVIDIA and is being provided under the terms and
# conditions of a form of NVIDIA software license agreement by and
# between NVIDIA and Licensee （“License Agreement“） or electronically
# accepted by Licensee.  Notwithstanding any terms or conditions to
# the contrary in the License Agreement reproduction or disclosure
# of the Licensed Deliverables to any third party without the express
# written consent of NVIDIA is prohibited.
#
# NOTWITHSTANDING ANY TERMS OR CONDITIONS TO THE CONTRARY IN THE
# LICENSE AGREEMENT NVIDIA MAKES NO REPRESENTATION ABOUT THE
# SUITABILITY OF THESE LICENSED DELIVERABLES FOR ANY PURPOSE.  IT IS
# PROVIDED “AS IS“ WITHOUT EXPRESS OR IMPLIED WARRANTY OF ANY KIND.
# NVIDIA DISCLAIMS ALL WARRANTIES WITH REGARD TO THESE LICENSED
# DELIVERABLES INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
# NONINFRINGEMENT AND FITNESS FOR A PARTICULAR PURPOSE.
# NOTWITHSTANDING ANY TERMS OR CONDITIONS TO THE CONTRARY IN THE
# LICENSE AGREEMENT IN NO EVENT SHALL NVIDIA BE LIABLE FOR ANY
# SPECIAL INDIRECT INCIDENTAL OR CONSEQUENTIAL DAMAGES OR ANY
# DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE DATA OR PROFITS
# WHETHER IN AN ACTION OF CONTRACT NEGLIGENCE OR OTHER TORTIOUS
# ACTION ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
# OF THESE LICENSED DELIVERABLES.
#
# U.S. Government End Users.  These Licensed Deliverables are a
# “commercial item“ as that term is defined at 48 C.F.R. 2.101 （OCT
# 1995） consisting of “commercial computer software“ and “commercial
# computer software documentation“ as such terms are used in 48
# C.F.R. 12.212 （SEPT 1995） and is provided to the U.S. Government
# only as a commercial end item.  Consistent with 48 C.F.R.12.212 and
# 48 C.F.R. 227.7202-1 through 227.7202-4 （JUNE 1995） all
# U.S. Government End Users acquire the Licensed Deliverables with
# only those rights set forth herein.
#
# Any use of the Licensed Deliverables in individual and commercial
# software must include in the user documentation and internal
# comments to the code the above Disclaimer and U.S. Government End
# Users Notice.
#

import os
import argparse
import numpy as np
import pycuda.driver as cuda
import tensorrt as trt

try:
    # Sometimes python2 does not understand FileNotFoundError
    FileNotFoundError
except NameError:
    FileNotFoundError = IOError

def GiB（val）:
    return val * 1 << 30

def find_sample_data（description=“Runs a TensorRT Python sample“ subfolder=““ find_files=[]）:
    ‘‘‘
    Parses sample arguments.
    Args:
        description （str）: Description of the sample.
        subfolder （str）: The subfolder containing data relevant to th

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2020-04-01 16:26  yolov3-tiny-onnx-TensorRT-master\
     文件         241  2020-04-01 16:04  yolov3-tiny-onnx-TensorRT-master\Predict.py
     文件        1739  2019-08-01 15:34  yolov3-tiny-onnx-TensorRT-master\README.md
     目录           0  2020-04-01 16:26  yolov3-tiny-onnx-TensorRT-master\__pycache__\
     文件        3631  2020-04-01 16:07  yolov3-tiny-onnx-TensorRT-master\__pycache__\common.cpython-36.pyc
     文件       11051  2020-04-01 16:07  yolov3-tiny-onnx-TensorRT-master\__pycache__\data_processing.cpython-36.pyc
     文件        6605  2020-04-01 16:26  yolov3-tiny-onnx-TensorRT-master\__pycache__\onnx_to_tensorrt.cpython-36.pyc
     文件         624  2019-08-01 15:34  yolov3-tiny-onnx-TensorRT-master\coco_labels.txt
     文件        6618  2019-08-01 15:34  yolov3-tiny-onnx-TensorRT-master\common.py
     文件       13653  2020-04-01 16:07  yolov3-tiny-onnx-TensorRT-master\data_processing.py
     目录           0  2019-08-01 15:34  yolov3-tiny-onnx-TensorRT-master\images\
     文件      163759  2019-08-01 15:34  yolov3-tiny-onnx-TensorRT-master\images\dog.jpg
     文件      141886  2019-08-01 15:34  yolov3-tiny-onnx-TensorRT-master\images\eagle.jpg
     文件      133495  2019-08-01 15:34  yolov3-tiny-onnx-TensorRT-master\images\horses.jpg
     文件          18  2019-08-01 15:34  yolov3-tiny-onnx-TensorRT-master\imageslist.txt
     文件        8582  2020-04-01 16:26  yolov3-tiny-onnx-TensorRT-master\onnx_to_tensorrt.py
     文件        8129  2020-04-01 16:07  yolov3-tiny-onnx-TensorRT-master\onnx_to_tensorrt.pyc
     文件          67  2019-08-01 15:34  yolov3-tiny-onnx-TensorRT-master\requirements.txt
     目录           0  2020-04-01 16:27  yolov3-tiny-onnx-TensorRT-master\result\
     文件      683962  2020-04-01 16:27  yolov3-tiny-onnx-TensorRT-master\result\dog_416_bboxes.png
     文件      621772  2020-04-01 16:27  yolov3-tiny-onnx-TensorRT-master\result\eagle_416_bboxes.png
     文件      569337  2020-04-01 16:27  yolov3-tiny-onnx-TensorRT-master\result\horses_416_bboxes.png
     文件        1916  2020-04-01 15:33  yolov3-tiny-onnx-TensorRT-master\yolov3-tiny.cfg
     文件    35447404  2020-04-01 15:34  yolov3-tiny-onnx-TensorRT-master\yolov3-tiny.onnx
     文件    19159712  2020-04-01 16:08  yolov3-tiny-onnx-TensorRT-master\yolov3-tiny.trt
     文件       34556  2020-04-01 15:31  yolov3-tiny-onnx-TensorRT-master\yolov3_to_onnx.py