- 博客(52)
- 收藏
- 关注
原创 Podman起服务时报错:Error: setting up CDI devices: unresolvable CDI devices nvidia.com/gpu=0
意思是:容器运行时(Docker 或 containerd)想通过 CDI 使用 GPU,但找不到名为的 CDI 设备。也就是:被要求使用 GPU 0,但 CDI 清单里根本没有这个设备。
2025-11-23 14:56:09
155
原创 Linux安装Nvidia驱动以及离线安装 NVIDIA Container Toolkit
将下载的文件传到服务器上分别运行sudo dpkg -i libnvidia-container1_1.17.8-1_amd64.deb。下载对应的版本的四个文件。
2025-11-23 14:49:50
337
原创 Triton部署YOLO,使用Python Backend
本文介绍了使用Triton部署YOLOv8/v11模型的方法,包括模型配置、前后处理流程和性能测试。主要内容包括:1)Triton模型目录结构配置,包含前处理、推理模型和后处理三个模块;2)详细的前后处理Python实现,包括图像预处理(letterbox缩放)、NMS后处理等;3)通过Ensemble配置将多个模块串联为完整流水线;4)提供Docker启动命令和Python客户端测试脚本,支持性能测试和结果可视化。该方法实现了端到端的YOLO模型部署,可通过配置文件动态调整推理参数,适用于实际生产环境。
2025-10-16 18:39:11
1176
原创 在Docker中pip离线安装python的各种包
docker cp /home/abc/tangrui_work/triton_packages_38 你的容器名:/tmp/docker commit 你的容器名 triton_python:v2.0。然后使用docker images看一下是否成功即可。docker exec -it 你的容名 bash。将提前下载好的python安装包复制到容器里。在容器安装后,然后提交成新镜像。
2025-10-15 15:55:32
224
原创 统一更改文件夹下图片命名
摘要 本文介绍了一个Python脚本,用于批量重命名文件夹中的图片文件。该脚本支持多种常见图片格式(如.jpg、.png等),允许用户自定义文件名前缀、起始序号和序号位数。脚本会自动检测图片文件,按顺序编号并重命名,同时避免文件名冲突。使用示例展示了如何将图片重命名为"new_001.jpg"、"new_002.jpg"等格式的序列文件。该工具适用于需要批量整理图片文件的场景。
2025-08-08 10:12:16
185
原创 U版RTDETR部署,使用C++通过ONNXRuntime推理
本文使用ONNX版本为1.14.1,OpenCv版本为4.8.0(同时也支持4.5.0,4.7.0),RTDETR不需要后处理所以不需要进行后处理,与YOLO不同的点还有YOLO需要特征金字塔所以输出的tenser尺寸是动态变动的,RTDETR默认输入300个框,所以Tenser在输出上默认为[1,300,class+4]。
2025-08-06 16:39:20
761
2
原创 YOLOv8/YOLOv11 C++ OpenCV DNN推理
本文介绍了使用C++和OpenCV部署YOLOv8目标检测模型的方法。首先需将YOLOv8的.pt模型转换为ONNX格式。代码实现了一个YOLO类,包含预处理、推理和后处理功能,支持多种输出格式解析,并实现了非极大值抑制(NMS)和结果可视化。主要步骤包括:加载ONNX模型、图像预处理、执行推理、解析输出张量、应用NMS过滤冗余框,最后绘制检测结果。该方案使用OpenCV DNN模块,支持CPU推理,适用于台球检测等场景,可输出带类别标签和置信度的检测框。
2025-08-01 10:47:05
567
2
原创 PYQT5多线程问题打包问题
如果pyqt界面中使用了多线程,在打包成exe后会出现点击多线程任务会重新加载整个界面,需要在打包为exe前在ui界面的main函数下加上multiprocessing.freeze_support()
2024-11-13 11:15:24
220
原创 【目标检测】根据检测模型推理结果构造一个XML文件
目标检测的数据集标注一般通过labelimg插件进行标注,有很多样本类型很容易进行标注对于这些很容易进行标注的数据集我们在进行模型迭代的时候经常也要进行标注这样导致很废时间,我们假设是在你已经有一个训练好基于某个任务的模型,这时你又有了新的训练样本那么如果不想费力再一个一个进行标注,可以通过当前已经训练好的模型线对这些未标记的样本进行推理再将每个样本的结果保存到xml文件中。将保存下来的xml文件使用labelimg插件就可以自动进行匹配。
2023-03-19 16:41:44
359
原创 【目标检测】根据检测模型的推理结果进行预测框的绘制
目前主流的目标检测器以YOLOX为例,后处理后的output基本上output[: , :4]前四个代表的bbox的坐标一般会将中心宽高的形式转换成左上角右下角点坐标的形式,至于output[: , :4] output[:, :5]分别是obj是否包含物体的得分和类别得分,但是目前例如YOLOv8已经没有obj得分了似乎只有分类得分大家根据情况对脚本进行更改。最后一个维度为类别索引.首先这里这里只是给大家提供一个脚本,具体从model出来的output需要经过的一些后处理例如nms自行解决。
2023-03-19 15:41:08
785
原创 【机器学习】利用DBSCAN-基于密度的空间聚类算法计算出一批经纬度中的中心点
【机器学习】利用DBSCAN-基于密度的空间聚类算法计算出一批经纬度中的中心点,并筛选出离群点。
2023-03-07 16:54:13
1626
1
原创 【PIL】OSError: cannot write mode RGBA as JPEG
如果你保存的图片是.png结尾通过重命名的方式更改为jpg会再进去存储的时候会报错,主要因为png图片默认有四个通道。PNG是四通道:RGBA意思是红色,绿色,蓝色,Alpha的色彩空间,Alpha指透明度。JPG是三通道:RGB意思是红色,绿色,蓝色。转换成rgb图片即可。
2023-03-06 16:40:39
663
原创 【pytorch】Pytorch训练好的模型转torchscript
TorchScript,它是PyTorch模型(子类nn.Module)的中间表示,可以在高性能环境(例如C ++)中运行。利用torch.jit.trace()函数。
2023-03-05 15:39:43
2146
4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅