TensorRT – 自带工具trtexec的参数使用说明
本文以TensorRT-7.2.3.4说明自带工具trtexec工具的使用参数进行说明。 1 trtexec的参数使用说明 === Model Options === --uff=<file> UFF model --onnx=<file> ONNX model --mode…
- TensorRT
- 2021-09-17
Pytorch – 使用torch.onnx.export将Pytorch模型导出为ONNX模型
1 torch.onnx.export torch.onnx.export( model, args, f, export_params=True, verbose=False, training=<TrainingMode.EVAL: 0>, input_names=None, out…
- Pytorch
- 2021-09-16
CUDA 安装报错 could not create file “…\chrome_elf.dll”
1 CUDA 安装报错 could not create file "...\chrome_elf.dll" 今天重新在win 7上安装CUDA 10.2的时候报了这个错,人都懵了,我以为是卸载之前的CUDA 10.0没有卸载干净导致注册表有残留信息,然后重启了电脑再安装,然后在解压exe的时候依然…
- 人工智能
- 2021-09-10
如何正确的选择深度学习模型工业化部署的方式
1 如何正确的选择深度学习模型工业化部署的方式 转自知乎高赞回答:https://www.zhihu.com/question/329372124/answer/743251971 最近在做深度学习模型的工业化部署,无意中看到了这篇高赞回答,觉得写得很好,有必要备份一下。这篇回答对我如何选择部署方式…
- 人工智能
- 2021-09-10
Mediapipe – 将Mediapipe handtracking封装成动态链接库dll/so,实现在桌面应用中嵌入手势识别功能
1 将Mediapipe handtracking封装成动态链接库dll/so的意义 Mediapipe提供了在cpu低延时下的高性能以及丰富功能,比如人脸识别,手部跟踪与识别,全身关节跟踪与识别,如果我们能在我们的应用中加入这些功能岂不是很酷! 在python环境下,我们可以通过直接安装Media…
- 姿态估计
- 2021-08-13
Mediapipe – Windows10 编译Mediapipe C++版本保姆级教程
1 Windows10编译Mediapipe C++版本 Win7我按照底下的步骤试了一下,在bazel编译的时候总是报: ERROR: SymlinkDirectories(C:/Users/Administrator/_bazel_Administrator/install/7e447298f5…
- 姿态估计
- 2021-08-09
Pytorch – 检测CUDA、cuDNN以及GPU版本的Pytorch是否安装成功、GPU显存测试
1 测试代码 import time import torch if __name__ == '__main__': print('torch版本:'+torch.__version__) print('cuda是否可用:'+str(torch.cuda.is_available())) print…
- Pytorch
- 2021-08-06
Mediapipe – 使用Mediapipe Holistic识别身体、手、面部全身关节点
1 Mediapipe & Holistic 1.1 Mediapipe MediaPipe 是一款由 Google Research 开发并开源的多媒体机器学习模型应用框架。 github地址:https://github.com/google/mediapipe 可使用以下命令安装: p…
- 姿态估计
- 2021-08-03
Alphapose – 在Alphapose中使用yolov3-tiny检测器大幅提升检测性能
1 Alphapose中使用yolov3-tiny检测器的应用场景 Alphapose的官方仓库中默认使用的是yolov3-spp,根据官网公布的数据,其mAP为60.6,FPS能到20FPS。但是如果我们面临的是一个不复杂的场景,场景中人数较少,而我们有需要其性能够好,实时性能达到应用的要求或者更…
- 姿态估计
- 2021-08-02
Python – list/numpy/pytorch tensor相互转换
1 list与numpy相互转换 list转numpy import numpy as np if __name__ == '__main__': a = [1,2,3,4] n = np.array(a) print(n) numpy转list import numpy as np if __na…
- Python
- 2021-07-26
yolov5 – 原始仓库检测代码detect.py中文注释
1 检测参数以及main函数解析 if __name__ == '__main__': """ weights:训练的权重 source:测试数据,可以是图片/视频路径,也可以是'0'(电脑自带摄像头),也可以是rtsp等视频流 output:网络预测之后的图片/视频的保存路径 img-size:网…
- 人工智能
- 2021-07-26
如何选择一块合适的用于深度学习的GPU/显卡
1 决定GPU性能的硬件参数 GPU架构 不同款的GPU可能采用不同设计架构,比如GeForce 10系列的GTX 1080/1080Ti采用的是Pascal架构,而GeForce 20系列的RTX 2080/2080Ti采用的是Turing架构。不同架构的GPU,即使其他参数差不多,性能差别可能非…
- 人工智能
- 2021-07-15