CUDA
关注
分享
扫一扫
文章平均质量分 51
cxb1998
这个作者很懒，什么都没留下…
展开
-
tensorRT部署实战——yolov5
yolov5的onnx简化到trt推理原创 2022-08-27 22:28:51 · 6945 阅读 · 4 评论 -
Warpaffine
yolo预处理等比缩放黑边填充中的缩放平移原创 2022-08-27 17:38:41 · 327 阅读 · 0 评论 -
tensorRT部署实战——分类器
tensorRT部署实例——resnet分类器原创 2022-08-22 21:38:59 · 1649 阅读 · 3 评论 -
onnx-trt
trt onnx介绍以及torch转onnx原创 2022-08-18 00:18:11 · 720 阅读 · 0 评论 -
CNN推理和动态shape
cnn的动态输入原创 2022-08-17 18:37:11 · 744 阅读 · 0 评论 -
TensorRT模型推理
trt推理原创 2022-08-17 16:34:11 · 1725 阅读 · 0 评论 -
TensorRT基础
用pytorch（pth）—> onnx —> trt的工作路径，可移植性高。下运行（变异的trt版本、cuda版本、编译时的GPU型号）1. TensorRT的核心在于对模型算子的优化（2. tensorRT的模型需要在目标GPU上。3. tensorRT生成的模型只能在。的方式选择最优算法和配置。...原创 2022-08-07 01:45:00 · 466 阅读 · 0 评论 -
NMS（非极大值抑制）的python，cpu，gpu实现
NMS（非极大值抑制）是目标检测中用来确定最佳检测框的手段，根据目标检测流程，若果没有NMS步骤，其每个检测框都会有大量重叠度很高的预测框表示同一个目标。如下图：左图为经过NMS的预测结果，右图为未经过NMS的结果，很明显，左图才是我们需要的结果。.........原创 2022-08-06 17:10:48 · 2578 阅读 · 1 评论 -
CUDA线程层级
CUDA线程层级原创 2022-08-04 20:09:56 · 424 阅读 · 0 评论 -
CUDA内存模型和线程束
CUDA内存模型原创 2022-08-04 20:09:14 · 100 阅读 · 0 评论