Spatio-Temporal MLP-Graph 复现过程【笔记】

昔柯的修炼日记

已于 2024-07-25 18:34:57 修改

阅读量756

点赞数 29

分类专栏：论文复现文章标签：笔记 python 计算机视觉机器学习

于 2024-04-15 14:46:39 首次发布

本文链接：https://blog.csdn.net/qq_43668996/article/details/136679485

版权

论文复现专栏收录该内容

1 篇文章 0 订阅

订阅专栏

小白式教程，超详细，复现全过程以及出现的问题和对应解决方法。在自己电脑上复现的，非服务器复现，不过两者步骤差别不大。

论文地址

项目代码地址

https://github.com/nies14/Spatio-Temporal-MLP-Graph/blob/main/README.md

一资源下载

下载数据集

https://download.csdn.net/download/qq_43668996/88957994

下载预训练模型

https://download.csdn.net/download/qq_43668996/88958989

保存在项目根目录下，如下图

二创建虚拟环境，配置pycharm

配置指令顺序、版本按项目文档中 READEME.md文件进行

1 打开Anaconda Prompt

2 输入 conda create -n mlpgraph python=3.8

输入 y

3 激活刚刚创建的环境，输入 conda activate mlpgraph ，前缀变为mlpgraph则激活成功

4 安装pytorch ，输入 pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

第一次安装时出现了以下错误，重新删除环境再安装又正常了............

报错 ERROR: Could not find a version that satisfies the requirement torch==1.7.0+cu110 (from versions: 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0, 1.12.0, 1.12.1, 1.13.0, 1.13.1, 2.0.0, 2.0.1, 2.1.0, 2.1.1, 2.1.2, 2.2.0, 2.2.1)
ERROR: No matching distribution found for torch==1.7.0+cu110

出现上述报错，可以直接点击提供的链接，找到我们需要的版本进行下载

版本名中的cp表示python版本

再根据文件地址进行安装。

5 配置 requirements.txt 文件（位置在项目代码文档里）。

直接输入指令 pip install -r requirements.txt的绝对路径（替换为自己的路径）

配置出现了错误，找不到匹配的torch。

这里需要我们进入requirements文件将torch行注释掉

再重新输入指令 pip install -r requirements.txt的绝对路径（替换为自己的路径）即可

6 配置pycharm

用Pycharm打开项目文档，点击右下角 Add New Interpreter ，再点击 Add Local Interpreter...

选择 Conda Environment , 再选择Use Existing ，再选择你anaconda安装目录下的conda.exe，j接着点击Load Environment ，接着你创建的项目的环境，最后点击OK，就完成啦！！！

三训练

训练指令 python main_graph.py -k hr --pro_train 1 --save_model 1 --save_dir './checkpoint/train-scratch/HR-NET/Frame-243/Pose-Refine' --post_refine --save_out_type post --show_protocol2 --nepoch 51 -z 384 -ds 384 -dc 768 --batchSize 256 --pad 121

打开main_graph 文件，选择 Edit Configurations....

将 -k hr --pro_train 1 --save_model 1 --save_dir './checkpoint/train-scratch/HR-NET/Frame-243/Pose-Refine' --post_refine --save_out_type post --show_protocol2 --nepoch 51 -z 384 -ds 384 -dc 768 --batchSize 256 --pad 121 粘贴到参数框，点击Apply、Run