TRT_object_detection:使用 TensorRT 引用对象检测模型的 Python 示例

时间:2024-06-19 16:30:32
【文件属性】:

文件名称:TRT_object_detection:使用 TensorRT 引用对象检测模型的 Python 示例

文件大小:28KB

文件格式:ZIP

更新时间:2024-06-19 16:30:32

Python

用于对象检测的 TensorRT Python 示例 性能包括 memcpy 和推理。 模型 输入尺寸 同仁纳米 ssd_inception_v2_coco(2017) 300x300 49毫秒 ssd_mobilenet_v1_coco 300x300 36ms ssd_mobilenet_v2_coco 300x300 46ms 由于预处理的优化还没有准备好,我们这里不包括图像读/写时间。 安装依赖 $ sudo apt-get install python3-pip libhdf5-serial-dev hdf5-tools $ pip3 install --extra-index-url https: // developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.13.1+nv19


【文件预览】:
TRT_object_detection-master
----README.md(3KB)
----main.py(3KB)
----coco.py(1KB)
----config()
--------model_ssd_mobilenet_v2_coco_2018_03_29.py(2KB)
--------model_ssd_inception_v2_coco_2017_11_17.py(2KB)
--------model_ssd_mobilenet_v1_coco_2018_01_28.py(2KB)
----lib()
--------libflattenconcat.so(73KB)

网友评论