1. 首页
  2. 人工智能
  3. 深度学习
  4. 使用Python和TensorRT进行rt-detr目标检测推理代码

使用Python和TensorRT进行rt-detr目标检测推理代码

上传者: 2023-12-03 16:15:12上传 PY文件 7.07KB 热度 68次

该代码展示了如何使用Python和TensorRT对rt-detr模型进行推理。用户可以输入图片或图片文件夹的路径以及模型路径,通过运行命令 python ./infer_tensorrt.py --infer_dir=./inputimgs/ --output_dir ./outputimgs/ --models ./rtdetr_hgnetv2_l_6x_coco.trt 来获取推理结果。这个过程实现了对单张图片或图片文件夹的批量推理,并将结果保存在指定的输出图片路径。

下载地址
用户评论