- 博客(7)
- 收藏
- 关注
RSS订阅原创 【TensorRT】记一次使用C++接口TensorRT部署yolov5 v6.1模型的过程
使用C++接口的TensorRT部署YOLOv5 v6.1模型
2022-08-02 12:43:53
5809
9
原创 保姆级教学如何在Ubuntu 20.04工作站上配置深度学习环境
在Ubuntu 20.04工作站上配置深度学习环境现在大多数深度学习攻城狮都会选择在云服务器上部署自己的深度学习环境，将其作为自己的深度学习工作站。在此我将介绍在Ubuntu 20.04 的深度学习工作站上配置深度学习环境的步骤。本工作站的配置为一块Tesla V100，CPU为8核16G。配置Ubuntu系统登录上服务器后，第一步为配置Ubuntu的系统。首先第一件事情肯定是给Ubuntu的软件管理器apt换国内镜像源。因为Ubuntu的apt源默认在国外，因此直接使用apt命令去下载软件速度非常
2022-02-11 11:19:07
3839
原创 TensorRTx-YOLOv5工程解读（二）
TensorRTx-YOLOv5工程解读（二）具体Workflow if (!wts_name.empty()) { IHostMemory* modelStream{ nullptr }; APIToModel(BATCH_SIZE, &modelStream, is_p6, gd, gw, wts_name); assert(modelStream != nullptr); std::ofstream p(engine_n
2021-10-07 11:26:01
1153
1
原创 TensorRTx-YOLOv5工程解读（一）
TensorRTx工程解读——YoloV5（一）权重生成：gen_wts.py作者先是使用了gen_wts.py这个脚本去生成wts文件。顾名思义，这个.wts文件里面存放的就是.pt文件的权重。脚本内容如下：import sysimport argparseimport osimport structimport torchfrom utils.torch_utils import select_devicedef parse_args(): parser = argpars
2021-09-26 15:30:35
2797
9
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人