pytorch复现stgcn
时间: 2023-10-28 10:07:04 浏览: 148
扩散模型pytorch复现
要复现STGCN,您可以按照以下步骤操作:
1. 首先,确保您已经安装了torch库的旧版本0.4.0。您可以使用以下命令进行安装:sudo pip3 install torch==0.4.0 -f https://download.pytorch.org/whl/cup/stable -i Simple Index
2. 接下来,您需要下载STGCN的代码。在终端中运行以下命令:
- 旧版:python3 main.py demo_old --video /home/chq/st-gcn/resource/media/test.mp4 --openpose /home/chq/openpose/build
- 新版STGCN:python3 main.py demo --openpose '/home/chq/openpose/build' --video '/home/chq/st-gcn/resource/media/skateboarding.mp4'
请注意,在运行这些命令之前,您需要将相应的视频和openpose路径替换为您自己的路径。
阅读全文