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
姿态估计 – Halpe Full-Body136数据集骨骼关节keypoint标注对应
1 Halpe Full-Body136数据集骨骼关节keypoint标注对应 Halpe Full-Body136数据集包含了面部(68个关节点)、身体(26个关节点)、左右手(各21个关节点),全身共计136个关节点。 项目地址:https://github.com/Fang-Haoshu/Ha…
- 姿态估计
- 2021-06-28
姿态估计 – COCO-WholeBody数据集骨骼关节keypoint标注对应
1 COCO-WholeBody数据集骨骼关节keypoint标注对应 COCO-WholeBody数据集注释了 4 种类型的边界框(人框、人脸框、左手框和右手框)和 133 个关键点(身体 17 个,脚部 6 个,面部 68 个,手部 42 个) 项目网址: https://github.com/…
- 姿态估计
- 2021-06-10
Pytorch – 使用opencv-python解码视频文件并将视频帧转换为Pytorch tensor作为网络模型输入数据
[toc] 1 视频文件作为网络模型的输入数据 越来越多的神经网络模型开始以视频作为训练数据,比如基于视频数据的行为识别等等,这就需要我们将视频转换为可适用的张量,本文将以pytorch为例,展示一下视频数据转换为pytorch tensor的过程。 2 使用OpenCV-Python解码视频文件并…
- Python
- 2021-04-19
Alphapose – Windows下Alphapose(Pytorch 1.1+)版本2021最新环境配置步骤以及踩坑说明
1 Alphapose github地址:https://github.com/MVIG-SJTU/AlphaPose Alphapose最初发布采用的是pytorch 0.4.0版本,不过现在使用的都是Pytorch 1.1+版本。 目前Pytorch 1.1+版本不像之前pytorch 0.4.…
- 姿态估计
- 2021-03-29
Pytorch – 使用Pyav解码视频文件并将视频帧转换为Pytorch tensor作为网络模型输入数据
1 视频文件作为网络模型的输入数据 越来越多的神经网络模型开始以视频作为训练数据,比如基于视频数据的行为识别等等,这就需要我们将视频转换为可适用的张量,本文将以pytorch为例,展示一下视频数据转换为pytorch tensor的过程。 2 使用Pyav解码视频文件并将视频帧转换为Pytorch …
- Pytorch
- 2021-03-25
我的开源项目 – 3DPoseEstimation从2D视频中估计人物三维姿势,并生成BVH文件
1 从2d视频中生成3d人物姿态,并导出为bvh文件 1.1 目的 从2d视频中检测人物2d关节点,并转换为三维空间坐标,然后使用dcm余弦矩阵计算关节的旋转信息并将其导出为bvh动作文件。 Github地址 : https://github.com/HW140701/VideoTo3dPoseAn…
- 姿态估计
- 2020-01-06