随笔分类 -  CUDA

TensorRT生成INT8校准文件
摘要：需要的资源 模型 训练模型时用的500~1000张图片 模型这里用的是onnx模型。 预处理 本文参考的例子是TensorRT的sampleSSD项目。 预处理需要将前述500张图片转成生成校准文件时需要的数据读取形式。 首先满足预处理代码中的convert命令，安装imagemagick apt- 阅读全文
posted @ 2022-10-18 07:57 yeren2046 阅读(805) 评论(6) 推荐(0) 编辑
基于CUDA查询显卡型号和显存大小
摘要：安装驱动后，我们可以用 nvidia-smi 命令来获取显卡信息。 nvidia-smi 的一个比较好的博客 nvidia-smi详解 nvidia-smi Tue Sep 27 11:08:37 2022 + + | NVIDIA-SMI 515.65.01 Driver Version: 515 阅读全文
posted @ 2022-09-27 20:47 yeren2046 阅读(873) 评论(0) 推荐(1) 编辑
nvjpeg 简单使用
摘要：nvjpeg是英伟达提供的一个可以将显存中的数据直接压缩成jpg图片的sdk，包含在cuda套件中。 官方接口文档 https://docs.nvidia.com/cuda/nvjpeg/index.html#nvjpeg-encode-image 一个写得挺好的博客 nvJPEG压缩图像， 本文代 阅读全文
posted @ 2022-09-21 10:32 yeren2046 阅读(2078) 评论(0) 推荐(0) 编辑
AV_PIX_FMT_CUDA 数据转 RGB
摘要：ffmpeg的nvdec解码结果数据格式为 AV_PIX_FMT_CUDA，实际使用中后续接算法需要转为RGB。算法跑在显卡上，解码也在显卡上，所以转换也定为直接在显卡上进行。 关于ffmpeg的nvdec解码的网上博客写的比较多，个人参考的一个比较好的 https://blog.csdn.net/ 阅读全文
posted @ 2022-09-21 00:22 yeren2046 阅读(517) 评论(1) 推荐(0) 编辑
CUDA JPEG编码
摘要：基于英伟达的jpegNPP工程，分离实现独立的JPEG压缩。 由于原工程是直接把解码时的jpeg图片的信息直接作为编码时的信息，所以在做独立的JPEG编码时，需要自己来填充各种信息。1.JPEG编码流程 从网上一片文章中解出来的图，红色框框中的流程图算是JPEG编码的一个流程图，对JPEG编码流程的了解有助于对代码的理解。2.Huffman表和量化表 Huffman表和量... 阅读全文
posted @ 2017-08-08 15:47 yeren2046 阅读(3799) 评论(0) 推荐(1) 编辑