Samples/python/yolov3_onnx/yolov3_to_onnx.py

AI & Data Science Deep Learning (Training & Inference) TensorRT
#1

In TensorRT samples/python/yolov3_onnx/yolov3_to_onnx.py
Why does it have the python 2 limitation? Is there tensorrt docker image with python2-cuda?

"""Run the DarkNet-to-ONNX conversion for YOLOv3-608."""
# Have to use python 2 due to hashlib compatibility
if sys.version_info[0] > 2:
    raise Exception("This script is only compatible with python2, please re-run this script with python2. The rest of this sample can be run with either version of python.")
#3

Hi @ding9801,
You can find the python 2 images in the below link


Thanks!
1 Like