Windows 下 YOLO26 的 TensorRT C++ 部署

简介: 本文详解Windows下基于TensorRT 10.16的YOLO26 C++部署全流程:涵盖VS2022/CUDA 12.9/OpenCV 4.8.1环境搭建、FP32/FP16/INT8模型导出与Engine序列化、命令行推理及自训练模型部署。实测单帧推理仅约1.4ms,速度较ultralytics Python提升14倍。

摘要:本文介绍如何在 Windows 下使用 TensorRT 10.16 以 C++ 部署 YOLO26 目标检测模型。内容涵盖环境准备(VS2022、CMake、CUDA 12.9、OpenCV 4.8.1、TensorRT)、工程编译与验证、模型导出(FP32/FP16/INT8)与 engine 序列化,以及命令行推理和部署自训练模型的完整流程。实测 C++ 实现推理单帧约 1.4 ms,相比 ultralytics Python 推理提升约 14 倍。

Windows 下 YOLO26 的 TensorRT C++ 部署

#

前言

这是一个系列教程,本片主要讲:YOLO26的目标检测,其他分类、分割、特征点、深度图都有。

仓库:TensorRT-Alpha
未来还会部署的模型包含SAM系列视觉LLM

在这里插入图片描述


仓库支持的模型矩阵

✅ 已跑通并实测 · 🚧 规划中 · — 未提供



| model | fp32 | fp16 | int8 | 序列化onnx |
|---|:---:|:---:|:---:|:---:|
| YOLO26 | | | | |
| yolo26-det | ✅ | ✅ | ✅ | ✅ |
| yolo26-seg | ✅ | ✅ | ✅ | ✅ |
| yolo26-obb | ✅ | ✅ | ✅ | ✅ |
| yolo26-cls | ✅ | ✅ | ✅ | ✅ |
| yolo26-depth | ✅ | ✅ | ✅ | ✅ |
| yolo26-sem | ✅ | ✅ | ✅ | ✅ |
| yolo26-pose | ✅ | ✅ | ✅ | ✅ |
| yolo26-tracking | 🚧 | 🚧 | 🚧 | 🚧 |
| YOLOV11 | | | | |
| yolov11_det | ✅ | ✅ | ✅ | ✅ |
| yolov11_seg | ✅ | ✅ | ✅ | ✅ |
| yolov11_obb | ✅ | ✅ | ✅ | ✅ |
| yolov11_cls | ✅ | ✅ | ✅ | ✅ |
| yolov11_pose | ✅ | ✅ | ✅ | ✅ |
| YOLOV9 | | | | |
| yolov9_det | ✅ | — | — | ✅ |
| yolov9_seg | ✅ | — | — | ✅ |
| YOLOV10 | | | | |
| yolov10_det | ✅ | — | — | ✅ |





其他(沿用 legacy 资产)



| model | fp32 | fp16 | int8 | 序列化onnx |
|---|:---:|:---:|:---:|:---:|
| yolov3 | ✅ | — | — | ✅ |
| yolov4 | ✅ | — | — | ✅ |
| yolov5 | ✅ | — | — | ✅ |
| yolov6 | ✅ | — | — | ✅ |
| yolov7 | ✅ | — | — | ✅ |
| yolov8_det | ✅ | ✅ | ✅ | ✅ |
| yolov8_seg | ✅ | ✅ | ✅ | ✅ |
| yolov8_cls | ✅ | ✅ | ✅ | ✅ |
| yolov8_pose | ✅ | ✅ | ✅ | ✅ |
| yolov8_obb | ✅ | ✅ | ✅ | ✅ |
| yolox | ✅ | — | — | ✅ |
| yolor | ✅ | — | — | ✅ |
| yolo_nas | ✅ | — | — | ✅ |
| efficientdet | ✅ | — | — | ✅ |
| yunet | ✅ | — | — | ✅ |
| u2net | ✅ | — | — | ✅ |




一、先看结果

1.1 速度

请添加图片描述

三组柱子,每组左边蓝色是这套 C++ 实现,右边橙色是 ultralytics 官方 Python 推理。测试片段是 1912×1080 的视频,imgsz 640,C++ 这边 batch=2 折算到每帧,Python 那边 batch=1:

阶段 Python 本仓库 倍数
预处理(5 步合一 kernel) 1.380 ms 0.043 ms 31.8×
推理 19.685 ms 1.398 ms 14.1×
后处理 0.446 ms 0.061 ms 7.3×
合计(不含搬运) 21.512 ms 1.502 ms 14.3×
合计(含 H2D) 21.512 ms 2.248 ms 9.6×

Python 那边是 CPU 版 torch,不是显卡没跑起来,是我故意用 CPU 版对拍的。拿 GPU 版 torch 比,后处理那栏它反而会好看一些(NMS 在 GPU 上做),预处理和推理还是差一个量级。

图上的柱子是纯计算,不含数据搬运。C++ 这边上传的是一整张 1912×1080 的原图,一帧 6.19 MB,走的还是 pageable 内存,光这一步就 0.68 ms,比预处理 kernel 本身贵十几倍。等后面接了 GStreamer 让解码直接落显存,这 0.68 ms 就没了,这也是下一阶段性能提升最大的一块。

还有一点得说明白:2.248 ms 折下来 445 fps,是算上那 0.68 ms 搬运的上限。真跑视频的时候达不到,瓶颈在解码线程,实测整条流水线大概 240 fps 上下(GPU 利用率只有三成左右,活都在 CPU 解码上)。别看到 445 就以为写进播放器能跑 445。

1.2 复现效果:本仓库 TensorRT vs ultralytics

请添加图片描述

左边是这套 C++ 实现跑出来的,右边是 ultralytics 官方 PyTorch 输出,框、置信度、掩码都是对齐的。

二、Windows环境准备

2.1 版本清单

组件 版本 说明
Visual Studio 2022(17.x) 工作负载勾“使用 C++ 的桌面开发”
CMake ≥ 3.25 工程里写死了这个下限
显卡驱动 跟 CUDA 12.9 配套 太旧的话 CUDA 装不上
CUDA Toolkit 12.9 只装 Toolkit,不用装驱动
OpenCV 4.8.1 官方 exe 自解压包
TensorRT 10.16.1.11 Windows amd64 zip 包

2.2 装 VS2022 和 CMake

从官网下 Community 版就行,安装器默认停在“工作负载”那一页,只需要勾一个:

请添加图片描述

就是图上红框那个“使用 C++ 的桌面开发”。右侧默认组件里已经带了 MSVC v143 和 Windows 10/11 SDK,不用额外加。装完重启一次。CMake 直接去官网下,版本要求 3.25 以上。

2.3 装 CUDA 12.9

NVIDIA 官网的 CUDA Toolkit Archive 页面找 12.9,下载 exe (local),三个多 G,装的时候选“自定义”,只留 CUDA 那一组组件,Visual Studio Integration 一定要勾上。

装完开个新的 PowerShell:

nvidia-smi
nvcc -V

nvidia-smi 右上角会显示驱动支持的最高 CUDA 版本,nvcc -V 打出 12.9 就对了。新开的窗口才能读到 CUDA_PATH,装完不重开窗口可能会说找不到,别急着怀疑装坏了。

2.4 解压 OpenCV 4.8.1

下载 opencv-4.8.1-windows.exe,双击,解压路径填:

D:\ThirdParty

不要勾 “Add to the system PATH”。这一步只是解压,不写注册表、不改环境变量。

解压完的目录长这样:

D:\ThirdParty\opencv\build\x64\vc16\
    ├── bin\      opencv_world481.dll 等
    └── lib\      opencv_world481.lib + OpenCVConfig.cmake

后面配置时 -DOpenCV_DIR 要指到 D:\ThirdParty\opencv\build\x64\vc16\lib 这一层,也就是 OpenCVConfig.cmake 所在的那层。指到 opencv 根目录会报 Could not find OpenCV,这是最常见的错。

vc16 是 VS2019/2022 的 ABI,MSVC 从 19.2 到 19.4 都是这个目录,VS2022 直接用没问题。

2.5 解压 TensorRT 10.16.1.11

下载 TensorRT-10.16.1.11.Windows.amd64.cuda-12.9.zip,解压到 D:\ThirdParty,得到:

D:\ThirdParty\TensorRT-10.16.1.11\
    ├── bin\      nvinfer_10.dll、trtexec.exe
    ├── include\  NvInfer.h
    └── lib\      nvinfer_10.lib、nvonnxparser_10.lib

同样不跑安装程序、不加 PATH,这个版本的 TensorRT 就是绿色包。OpenCV 和 TensorRT 都别手动往系统环境变量里塞,后面 DeployRuntime.cmake 会把 dll 拷到 exe 旁边。


三、编译

3.1 拉代码

cd D:\VS2022_Project
git clone -b v2-dev https://github.com/FeiYull/TensorRT-Alpha.git
cd TensorRT-Alpha

路径按你自己习惯改,后面所有命令里的 D:\VS2022_Project\TensorRT-Alpha 都跟着换。

3.2 配置:cmake一把,建立vs工程

下面指令,换成你的实际路径就行!

cmake -S D:\VS2022_Project\TensorRT-Alpha `
      -B D:\VS2022_Project\TensorRT-Alpha\build `
      -G "Visual Studio 17 2022" -A x64 `
      -DOpenCV_DIR=D:/ThirdParty/opencv/build/x64/vc16/lib `
      -DTensorRT_ROOT=D:/ThirdParty/TensorRT-10.16.1.11

配置跑完,最后几行应该有这些:

-- [CUDA] arch detected from local GPU: 120
-- [OpenCV] version : 4.8.1
-- [OpenCV] dir     : D:/ThirdParty/opencv/build/x64/vc16/lib
-- [TensorRT] root       : D:/ThirdParty/TensorRT-10.16.1.11
-- [TensorRT] nvinfer    : .../nvinfer_10.lib
-- [TensorRT] onnxparser : .../nvonnxparser_10.lib
-- [TensorRT] CUDA       : 12.9 @ ...

如下图:

请添加图片描述

3.3 编译

可以用vs2022直接打开编译:

请添加图片描述

也可以使用指令直接编译

cmake --build build --config Release -j 24

编完的产物:

build\bin\Release\trt_alpha.exe

旁边会多出一堆 dll。这是 cmake/DeployRuntime.cmake 干的活:编译完自动把 OpenCV、TensorRT、CUDA 运行时的 dll 拷到 exe 旁边。因为 Windows 加载器找 dll 的第一优先级就是 exe 所在目录,所以不用给 PATH 加任何东西。

目录里长这样:

请添加图片描述

trt_alpha.exe 旁边是 opencv_world、nvinfer_10、nvinfer_plugin_10、nvonnxparser_10 这些 dll,再往下是 samples 和 test 编出来的一堆 exe,跑单测用的。

3.4 验证

.\build\bin\Release\trt_alpha.exe list

打出 21 个模型名,说明注册表、静态库、链接全通了:

Registered models (21):
  efficientdet
  u2net
  yolo_nas
  yolor
  yolov3
  yolov4
  yolov5
  yolov6
  yolov7
  yolov8
  yolov8_pose
  yolov8_seg
  yolov26
  yolov26_cls
  yolov26_depth
  yolov26_obb
  yolov26_pose
  yolov26_seg
  yolov26_sem
  yolox
  yunet

list 打出来的名字就是 --net 能接受的值,同时也决定默认读 configs/<名字>.ini。比如 --net yolov26_seg 读的就是 configs/yolov26_seg.ini。

3.5 编译期常见错误

报错 原因 处理
Could not find OpenCV OpenCV_DIR 指错层 指到 ...\build\x64\vc16\lib
Could not find TensorRT TensorRT_ROOT 里没有 include/NvInfer.h 检查是否多套了一层目录
C1083: 无法打开 cuda_runtime_api.h 没装 CUDA,或装完没重开窗口 echo $env:CUDA_PATH 是否有值
MSB6001: 字典中已添加相同键 环境里同时有 HTTPS_PROXY 和 https_proxy 两个键 跑 CMake 之前先把小写那份清掉,Git Bash 里 unset https_proxy,PowerShell 里 Remove-Item Env:https_proxy;这机器上确实会撞
未找到 nvinfer_10.lib TensorRT 还是 8.x 的包 TRT 10 的库名带 _10 后缀,装错版本了

四、准备模型

4.0 直接下载模型

onnx文件可以自己导出,也可以直接下载我提供的,在页面链接最下:
https://github.com/FeiYull/TensorRT-Alpha/releases/download/v1.0.0/yolo26n.onnx
或者你也可以按照如下步骤自己导出模型。

4.1 手动导出onnx

Python 的 yolo 环境,一句指令的事情:

pip install ultralytics

# FP32,精度对齐用这个
yolo export model=yolo26n.pt format=onnx imgsz=640 batch=2 opset=17 simplify=True

导出文件默认落在权重旁边,叫 yolo26n.onnx。

4.2 量化导出(FP16 / INT8)

注意:tensorrt10.x版本后续会删除量化工具,官方建议是:用户自己在pytorch端自行量化,然后导出量化版本的onnx,最后直接用tensorrt进行序列化就行(这条指令后续会说)。但是,这里我还是会说,最简单的,也是最不靠谱的PTQ量化方式如下,utralyrics更加最简QAT量化方法,自己可以搜下(QAT现在是业界主流):

yolo export model=yolo26n.pt format=onnx imgsz=640 batch=2 opset=17 simplify=True quantize=16

文件名还是 yolo26n.onnx,跟前一份重名,自己改成 yolo26n_fp16.onnx。

INT8 要校准集,导出的是 Q/DQ 图,文件名自动带 _int8 后缀:

yolo export model=yolo26n.pt format=onnx imgsz=640 batch=2 opset=17 simplify=True quantize=8 data=coco8.yaml

data 给一份 yolo 格式的 yaml,校准用的就是里面的图。coco8 只有 8 张,够跑通流程,真要上生产还是拿自己的数据集校准。

4.3 转成 engine

trtexec.exe 在 TensorRT 的 bin 目录里。先切过去,再执行:

cd D:\ThirdParty\TensorRT-10.16.1.11\bin

# FP32,精度对齐用这个
.\trtexec.exe --onnx=D:\models\yolo26n.onnx --saveEngine=D:\VS2022_Project\TensorRT-Alpha\engines\yolov26n.trt --noTF32

--noTF32 是给 FP32 引擎用的,把 TF32 关掉,这样 fp32 结果不会漂。只有精度基准才需要它。

FP16 更快,引擎也小一圈。量化本身在 Python 端就做完了,这边只是换个 flag:

.\trtexec.exe --onnx=D:\models\yolo26n.onnx --saveEngine=D:\VS2022_Project\TensorRT-Alpha\engines\yolov26n_fp16.trt --fp16

4.4 引擎文件放哪

configs/yolov26*.ini 里已经写好了各自的引擎路径,文件名对得上就不用改:

configs/yolov26.ini        -> engines/yolov26n.trt

引擎想放别处也行,两种办法:改 ini 里的 engine =,或者运行时 --engine 指过去。--engine 优先级最高,会盖掉 ini 和 --config。


五、跑起来

5.1 目录准备

TensorRT-Alpha\
├── build\bin\Release\trt_alpha.exe
├── configs\        不用动
├── data\           测试图,bus.jpg / zidane.jpg / road0.png 这些仓库里自带
└── engines\        把 .trt 放进来

5.2 检测

可以直接使用命令行,也可以使用vs2022打开工程,设置如图的sample_infer工程为启动。如图绿色框,然后设置trt文件路径,在设置一个视频,或者图片,或者路径(路径里面是多张图片),还可以是url、camera id等等

请添加图片描述

也可以直接在命令行执行推理

.\build\bin\Release\trt_alpha.exe run --net yolov26 --source data/bus.jpg  --engine d:/xxxx.trt   --save

--show 弹窗显示,窗口标题是 trt_alpha:<模型名>,同时开几个模型也不会互相覆盖。--save 落盘,不带参数时存到 save\<模型名>\,带路径就原样用那个路径。

请添加图片描述

5.3 参数source

只有一个 --source 参数,类型是程序自己判的:

写法 当成
data/bus.jpg 单张图
data/ 图片目录,逐张推理
data/people.mp4 本地视频
rtsp://192.168.1.10:554/stream1 网络流,rtsp / rtmp / http(s) / udp 都认
0 0 号摄像头

其他参数,这里不多说,自己探索。

六、部署自己训练的模型

训练部署一个模型:假设我们使用utralyrics训练一个目标检测模型(使用别的框架训练也是同理),有四个类别,

6.1、训练

训练代码如下:

from ultralytics import YOLO

def main():
    model = YOLO("yolo26n.pt")
    results = model.train(data="african-wildlife.yaml", epochs=100, imgsz=640)

if __name__ == '__main__':
    main()

训练好之后,得到权重文件:my_det_task_cls_num_4.pt

6.2、导出onnx

from ultralytics import YOLO

model = YOLO(r"runs\detect\train-3\weights\my_det_task_cls_num_4.pt")
model.export(
    format="onnx",
    batch = 2, # 仓库代码支持多batch推理,这里设置batch为2
    imgsz=640,
    dynamic=False, 
    simplify=True,
    opset=17,
    nms=None,      # 用默认 one-to-many 头,one-to-to精度略有下降,并且本仓库的decode+nms使用cuda c实现,足够快
)

然后导出对应的onnx,得到:my_det_task_cls_num_4.onnx

6.3、使用trtexec序列化

.\trtexec.exe --onnx=my_det_task_cls_num_4.onnx --saveEngine=my_det_task_cls_num_4.trt --noTF32

得到:my_det_task_cls_num_4.trt文件

6.4、新建ini格式配置文件

在仓库TensorRT-Alpha目录下

  1. 将上述my_det_task_cls_num_4.trt拷贝到engines文件夹下

  2. 在data/classes/目录下新建:african-wildlife.txt,用于设置我们训练时候的任务类别,我们这里是4个类别,具体类型信息如下:

    注:buffalo 255 0 255分别表示:name of label, B, G, R;其中BGR适用于渲染时的颜色。

    buffalo 255 0 255
    elephant 255 0 0
    rhino 0 255 255
    zebra 0 255 0
    
  3. configs文件夹下新建:my_det_task_cls_num_4.ini

具体如下:注意根据情况修改:engine、class_names_file、num_class、batch_size等变量

[model]
engine = engines/my_det_task_cls_num_4.trt
class_names_file = data/classes/african-wildlife.txt
num_class = 4

[input]
layout = nchw
batch_size = 2

[postprocess]
conf_thresh = 0.25
iou_thresh = 0.7
top_k = 300

6.5、执行

记住:任何时候,加载自己的模型,执行的时候,别忘记设定参数:--config configs/my_det_task_cls_num_4.ini,其他参数,类似engine的路径设置,都可以在ini文件中设置。

.\build\bin\Release\trt_alpha.exe run --source E:/Code/datasets/african-wildlife/images/test   --net yolov26 --config configs/my_det_task_cls_num_4.ini --engine engines/my_det_task_cls_num_4.trt  --save

以下是成功执行log:

(base) PS D:\VS2022_Project\TensorRT-Alpha>  .\build\bin\Release\trt_alpha.exe run --source E:/Code/datasets/african-wildlife/images/test   --net yolov26 --config configs/my_det_task_cls_num_4.ini --engine engines/my_det_task_cls_num_4.trt  --save
[2026-10-19 19:21:36.219] [INFO ] [tid=7600] [paths.cpp:208 root] Paths: root = D:\VS2022_Project\TensorRT-Alpha  (source: executable location)
[2026-10-19 19:21:36.219] [INFO ] [tid=7600] [config.cpp:174 loadModelConfig] Config: loaded configs/my_det_task_cls_num_4.ini (with configs/base.ini) (engine=engines/my_det_task_cls_num_4.trt, batch=2, layout=NCHW, extras=34 entries)
[2026-10-19 19:21:36.220] [INFO ] [tid=7600] [config.cpp:264 loadClassNamesFile] Config: class names file data/classes/african-wildlife.txt (4 classes)
[2026-10-19 19:21:36.220] [INFO ] [tid=7600] [inference_pool.cpp:50 InferencePool] [InferencePool] constructing with 1 worker(s), engine=engines/my_det_task_cls_num_4.trt, batchSize=2, maxQueueSize=4
[2026-10-19 19:21:36.220] [INFO ] [tid=7600] [inference_pool.cpp:55 InferencePool] [InferencePool] loading shared engine: engines/my_det_task_cls_num_4.trt
[2026-10-19 19:21:36.220] [INFO ] [tid=7600] [engine.cpp:381 Engine] Engine: loading D:\VS2022_Project\TensorRT-Alpha\engines\my_det_task_cls_num_4.trt
[TRT INFO ] Loaded engine size: 12 MiB
[2026-10-19 19:21:36.331] [INFO ] [tid=7600] [engine.cpp:452 discoverIo] Engine: io[0] input  'images' fp32 [2, 3, 640, 640]
[2026-10-19 19:21:36.331] [INFO ] [tid=7600] [engine.cpp:462 discoverIo] Engine: io[0] input batch static, min/opt/max = 2/2/2
[2026-10-19 19:21:36.331] [INFO ] [tid=7600] [engine.cpp:452 discoverIo] Engine: io[1] output 'output0' fp32 [2, 8, 8400]
[2026-10-19 19:21:36.332] [INFO ] [tid=7600] [engine.cpp:410 Engine] Engine: loaded (2 io tensors)
[2026-10-19 19:21:36.332] [INFO ] [tid=7600] [yolov26.cpp:72 loadConfig] yolov26: config num_class=4 extra=0 conf=0.25 iou=0.7 top_k=300
[TRT INFO ] [MS] Running engine with multi stream info
[TRT INFO ] [MS] Number of aux streams is 3
[TRT INFO ] [MS] Number of total worker streams is 4
[TRT INFO ] [MS] The main stream provided by execute/enqueue calls is the first worker stream
[TRT INFO ] [MemUsageChange] TensorRT-managed allocation in IExecutionContext creation: CPU +0, GPU +52, now: CPU 0, GPU 62 (MiB)
[2026-10-19 19:21:36.394] [INFO ] [tid=7600] [engine.cpp:372 applyInputShape] InputShape: 'images' layout=NCHW -> [2, 3, 640, 640]
[2026-10-19 19:21:36.395] [INFO ] [tid=7600] [yolov26.cpp:230 init] yolov26: initialized (batch=2, outC=8, anchors=8400, dst=640x640)
[2026-10-19 19:21:36.395] [INFO ] [tid=7600] [inference_pool.cpp:137 InferencePool] [InferencePool] worker[0] ready
[2026-10-19 19:21:36.395] [INFO ] [tid=7600] [inference_pool.cpp:147 InferencePool] [InferencePool] ready with 1 worker(s)
+===================================================================================================+
| [CONFIG] yolov26                                                configs/my_det_task_cls_num_4.ini |
| [model]                                                                                           |
|   engine            = engines/my_det_task_cls_num_4.trt                 CLI                       |
|   class_names_file  = data/classes/african-wildlife.txt                 my_det_task_cls_num_4.ini |
|   num_class         = 4                                                 my_det_task_cls_num_4.ini |
|                                                                                                   |
| [input]                                                                                           |
|   batch_size        = 2                                                 my_det_task_cls_num_4.ini |
|   layout            = nchw                                              my_det_task_cls_num_4.ini |
|                                                                                                   |
| [normalize]                                                                                       |
|   scale             = 255.0                                             base.ini                  |
|   mean              = 0.0,0.0,0.0                                       base.ini                  |
|   std               = 1.0,1.0,1.0                                       base.ini                  |
|   pad_value         = 114.0                                             base.ini                  |
|                                                                                                   |
| [postprocess]                                                                                     |
|   conf_thresh       = 0.25                                              my_det_task_cls_num_4.ini |
|   iou_thresh        = 0.7                                               my_det_task_cls_num_4.ini |
|   top_k             = 300                                               my_det_task_cls_num_4.ini |
|                                                                                                   |
| [pool]                                                                                            |
|   workers           = 1                                                 base.ini                  |
|   max_queue_size    = 4                                                 base.ini                  |
|   result_queue_size = 8                                                 base.ini                  |
|   render_threads    = 6                                                 base.ini                  |
|                                                                                                   |
| [output]                                                                                          |
|   jpeg_quality      = 95                                                base.ini                  |
|   save_dir          = save/yolov26                                      default                   |
|                                                                                                   |
| [resolved]   engine truth (shape = engine declaration; -1 = dynamic axis)                         |
|   batch             = 2                                                 (engine-confirmed)        |
|   in                = images   2 x 3 x 640 x 640   fp32 (4B)   linear                             |
|   out               = output0   2 x 8 x 8400   fp32 (4B)   linear                                 |
+===================================================================================================+
[2026-10-19 19:21:36.414] [INFO ] [tid=7600] [opencv_source.cpp:129 openImage] OpenCVSource: 'E:/Code/datasets/african-wildlife/images/test' is a directory -> images mode (single level)
[2026-10-19 19:21:36.415] [INFO ] [tid=7600] [opencv_source.cpp:165 openImages] OpenCVSource: found 227 image(s) in E:/Code/datasets/african-wildlife/images/test
[2026-10-19 19:21:36.415] [INFO ] [tid=7600] [opencv_source.cpp:96 OpenCVSource] OpenCVSource[images]: opened 'E:/Code/datasets/african-wildlife/images/test' batch=2 sourceId=0
[2026-10-19 19:21:36.422] [INFO ] [tid=7600] [pipeline.cpp:176 validateConfig] Pipeline: render_threads=6 (frame order on disk is not guaranteed)
[2026-10-19 19:21:36.422] [INFO ] [tid=7600] [pipeline.cpp:188 start] Pipeline: starting with 1 source(s) and 1 pool(s)
[2026-10-19 19:21:36.422] [INFO ] [tid=7600] [pipeline.cpp:200 start] Pipeline: launching 6 render thread(s)
[2026-10-19 19:21:36.423] [INFO ] [tid=7600] [pipeline.cpp:206 start] Pipeline: render thread(s) launched
[2026-10-19 19:21:36.423] [INFO ] [tid=7600] [pipeline.cpp:217 start] Pipeline: launching source thread 0
[2026-10-19 19:21:36.423] [INFO ] [tid=7600] [pipeline.cpp:220 start] Pipeline: all source threads launched
[2026-10-19 19:21:36.423] [INFO ] [tid=46752] [pipeline.cpp:338 renderLoop] Pipeline: render loop started, entering while
[2026-10-19 19:21:36.423] [INFO ] [tid=7600] [pipeline.cpp:225 start] Pipeline: started (1 source + 6 render = 7 threads)
[2026-10-19 19:21:36.423] [INFO ] [tid=48696] [pipeline.cpp:338 renderLoop] Pipeline: render loop started, entering while
[2026-10-19 19:21:36.423] [INFO ] [tid=31172] [pipeline.cpp:338 renderLoop] Pipeline: render loop started, entering while
[2026-10-19 19:21:36.423] [INFO ] [tid=49896] [pipeline.cpp:338 renderLoop] Pipeline: render loop started, entering while
[2026-10-19 19:21:36.423] [INFO ] [tid=49208] [pipeline.cpp:338 renderLoop] Pipeline: render loop started, entering while
[2026-10-19 19:21:36.423] [INFO ] [tid=34192] [pipeline.cpp:338 renderLoop] Pipeline: render loop started, entering while
[2026-10-19 19:21:36.423] [INFO ] [tid=24136] [pipeline.cpp:237 sourceLoop] Pipeline: source[0] thread entered
[2026-10-19 19:21:36.423] [INFO ] [tid=24136] [pipeline.cpp:243 sourceLoop] Pipeline: source[0] 'images' using pool[0]
[2026-10-19 19:21:36.434] [INFO ] [tid=46752] [opencv_renderer.cpp:979 save] OpenCVRenderer: saving into D:\VS2022_Project\TensorRT-Alpha\save\yolov26
[2026-10-19 19:21:36.434] [WARN ] [tid=46752] [opencv_renderer.cpp:1037 save] OpenCVRenderer::save: overwriting existing file(s), e.g. save/yolov26\1 (103).jpg (reported once per output dir)
[2026-10-19 19:21:37.188] [INFO ] [tid=24136] [pipeline.cpp:259 sourceLoop] Pipeline: source[0] next() returned false, breaking
[2026-10-19 19:21:37.188] [INFO ] [tid=24136] [pipeline.cpp:320 sourceLoop] Pipeline: source[0] exiting
[2026-10-19 19:21:37.194] [INFO ] [tid=49896] [pipeline.cpp:350 renderLoop] Pipeline: render loop pop() returned false, exiting
[2026-10-19 19:21:37.194] [INFO ] [tid=49896] [pipeline.cpp:494 renderLoop] Pipeline: render loop exiting
[2026-10-19 19:21:37.196] [INFO ] [tid=46752] [pipeline.cpp:350 renderLoop] Pipeline: render loop pop() returned false, exiting
[2026-10-19 19:21:37.196] [INFO ] [tid=46752] [pipeline.cpp:494 renderLoop] Pipeline: render loop exiting
[2026-10-19 19:21:37.198] [INFO ] [tid=48696] [pipeline.cpp:350 renderLoop] Pipeline: render loop pop() returned false, exiting
[2026-10-19 19:21:37.198] [INFO ] [tid=48696] [pipeline.cpp:494 renderLoop] Pipeline: render loop exiting
[2026-10-19 19:21:37.201] [INFO ] [tid=49208] [pipeline.cpp:350 renderLoop] Pipeline: render loop pop() returned false, exiting
[2026-10-19 19:21:37.201] [INFO ] [tid=49208] [pipeline.cpp:494 renderLoop] Pipeline: render loop exiting
[2026-10-19 19:21:37.204] [INFO ] [tid=31172] [pipeline.cpp:350 renderLoop] Pipeline: render loop pop() returned false, exiting
[2026-10-19 19:21:37.204] [INFO ] [tid=31172] [pipeline.cpp:494 renderLoop] Pipeline: render loop exiting
[2026-10-19 19:21:37.205] [INFO ] [tid=34192] [pipeline.cpp:350 renderLoop] Pipeline: render loop pop() returned false, exiting
[2026-10-19 19:21:37.205] [INFO ] [tid=34192] [pipeline.cpp:494 renderLoop] Pipeline: render loop exiting
[2026-10-19 19:21:37.205] [INFO ] [tid=7600] [pipeline.cpp:567 waitForCompletion] Pipeline: all threads joined
[2026-10-19 19:21:37.205] [INFO ] [tid=7600] [inference_pool.cpp:370 shutdown] [InferencePool] shutdown requested, queue=0, active=0
[2026-10-19 19:21:37.205] [INFO ] [tid=24208] [inference_pool.cpp:238 workerLoop] [InferencePool] worker[0] exiting
[2026-10-19 19:21:37.210] [INFO ] [tid=7600] [inference_pool.cpp:391 shutdown] [InferencePool] shutdown complete
(base) PS D:\VS2022_Project\TensorRT-Alpha>

以下是推理结果,效果图,分别是本仓库tensorrt推理效果和pytorch推理效果:

请添加图片描述
请添加图片描述

注:推理的时候设置rect=False,表示关闭矩形推理,矩形推理在tensorrt推理没啥吊用!

from ultralytics import YOLO
# 加载经过 COCO 预训练的 YOLO26n 模型
#model = YOLO("yolo26n-cls.pt")
#model = YOLO("yolo26n-cls.pt")
#model = YOLO("yolo26n-obb.pt")
#model = YOLO("yolo26n-pose.pt")
# model = YOLO("yolo26n-seg.pt")
#model = YOLO("yolo26n-depth.pt")
model = YOLO(r"runs\detect\train-3\weights\best.pt")
# 使用 YOLO26n 模型对 'bus.jpg' 图像进行推理
results = model(
    r"E:/Code/datasets/african-wildlife/images/test",
    rect=False,
    save=True,
    exist_ok=True           # 覆盖已有文件夹
)

都看到这里,去点个star吧。😄

相关文章
|
5天前
|
云安全 人工智能 运维
当攻击全面Agent化,你的安全防御能否跟上AI Speed?
多起客户侧AK泄露事件应急的真实观察
1242 4
|
28天前
|
人工智能 JSON API
全网刷屏的 Jev 模型正式开放!一手实战测评 + 保姆级教程
全网爆火的 Jev 模型是什么?有什么用?怎么使用?怎么接入 AI 编程工具?效果真的好么?傻子可懂的 Jev 保姆级实战教程 + 项目实战测评来啦
9791 32
|
8天前
|
存储 人工智能 安全
千问办公官网入口网页版登录:qwenwork.cn 注册送2000积分、登录送100积分
阿里云千问办公(QwenWork)是AI智能工作平台,支持一句话生成PPT、表格、网页、视频及数据分析。个人免费版送2000积分+每日登录赠100分,含1GB存储、5个发布页;企业版198元/席/月起。阿里千问办公QwenWork官网:https://t.aliyun.com/U/0VCTGt 阿里AI工作平台,一句话完成数据分析、PPT 生成、视频剪辑、网页搭建等复杂任务
|
3天前
|
人工智能 API 开发工具
DeepSeek Harness 神插件:白嫖 13 家大厂模型
`dsh-our-free-model` 是 DeepSeek Harness 神级插件:零配置、零API Key、零账号,聚合13家大厂免费模型(含DeepSeek V4.1 Flash、Kimi K3等),支持本地网关、用量看板与自动降级,真正实现“白嫖即用”。
|
14天前
|
存储 人工智能 并行计算
大模型本地部署终端选型方法论:以 Qwen3.8-27B 为例的四档分层完整流程
本文提出一套大模型本地部署终端选型方法论:定约束、定档位、定框架、定参数四步决策法,配合入门、主力、质量、无损四档分层模型。以 Qwen3.8-27B 实测数据为例,逐环节解读显存、带宽、存储、散热、系统、预算等要素,给出面向不同预算的优选方案、决策自查清单与市场观察框架。文末前瞻 AI 笔记本的 CPU+GPU 与统一内存两条路线,论证四步决策法在新品类上的延续性。
|
5天前
|
Web App开发 人工智能 JavaScript
太猛了,一个月干到15W+,给大家推荐一个最近爆火的开源逆向工具:REA
REA(Reverse Engineer Anything)是近期爆火的开源AI逆向工具,4月发布,1个月内斩获15万+ Star。它让AI Agent无需源码即可分析二进制、APK、JS/Electron、HAR包等17类目标,自动反编译、生成可运行代码,并强制返回带证据与局限的结论。全本地执行,MIT协议,支持Hopper/Ghidra/IDA等引擎,一行命令即可上手。
629 1
太猛了,一个月干到15W+,给大家推荐一个最近爆火的开源逆向工具:REA
|
13天前
|
人工智能 JSON 自然语言处理
2026 年 Jev 决策模型深度拆解:原理解读、实战测评与保姆级落地教程
有一款特殊AI模型在开发者圈子刷屏,它摒弃传统大模型擅长的对话聊天能力,专注做高速结构化决策,它就是TypeSafe AI推出的Jev模型。该模型由ChatGPT共同发明人Diogo Almeida主导研发,定位为**System One Model(系统一模型)**,对标人类大脑快速直觉判断的思维模式,在响应延迟、调用成本、结构化输出稳定性上相比传统生成式大模型有着巨大差异。本文会完整拆解Jev底层原理、三大核心原语能力、适用业务场景,同时提供可直接运行的curl、Python代码示例,并且结合多组实测数据,客观分析模型优势与能力边界,帮助普通开发者和AI应用从业者快速上手落地。
933 1
|
9天前
|
人工智能 JSON 自然语言处理
2026 年 Jev 决策模型全解析:核心原理拆解,性能实战测评与落地部署教程
有一款特殊AI模型在开发者圈子刷屏,它摒弃传统大模型擅长的对话聊天能力,专注做高速结构化决策,它就是TypeSafe AI推出的Jev模型。该模型由ChatGPT共同发明人Diogo Almeida主导研发,定位为System One Model(系统一模型),对标人类大脑快速直觉判断的思维模式,在响应延迟、调用成本、结构化输出稳定性上相比传统生成式大模型有着巨大差异。本文会完整拆解Jev底层原理、三大核心原语能力、适用业务场景,同时提供可直接运行的curl、Python代码示例,并且结合多组实测数据,客观分析模型优势与能力边界,帮助普通开发者和AI应用从业者快速上手落地。
594 0

热门文章

最新文章