pytorch_Realtime_Multi-Person_Pose_Estimation项目学习

提示：该文章仅仅记录自己的学习过程，如有问题，请大家指教。

项目下载地址：GitHub - tensorboy/pytorch_Realtime_Multi-Person_Pose_Estimation

一、环境配置

requirements.txt ——该文件提供了需要的库

1.torch——进入官网  ：Start Locally | PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/locally/根据自己电脑配置下载符合自己的pytorch

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

会遇到下载不了的情况——————换源

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - defaults
show_channel_urls: true

 

2.pycocotools的安装（仅适用于windows）

通过在清华镜像源中找到适合自己的版本下载下来

镜像地址：Links for pycocotools-windows

在虚拟环境下安装：pip install 文件名 

二、demo文件夹

picture_demo.py  输入图片关键点就会标注出来
web_demo.py  目前未学习

三、train文件夹

1._init_paths.py 用于实现路径的添加

2.train_Vgg19.py的学习（训练VGG19模型）

2.1从lib包中引用相关的模块

from lib.network.rtpose_vgg import get_model, use_vgg
from lib.datasets