stgcn pytorch复现
时间: 2023-09-30 12:07:13 浏览: 227
时空预测模型PyTorch复现
要复现ST-GCN模型,你需要按照以下步骤进行操作:
1. 首先,确保你已经安装了torch库的0.4.0版本。你可以使用以下命令在终端中安装:
```
sudo pip3 install torch==0.4.0 -f https://download.pytorch.org/whl/cup/stable -i Simple Index
```
2. 如果你使用的是旧版ST-GCN,请使用以下命令运行模型,并指定视频和Openpose路径:
```
python3 main.py demo_old --video /home/chq/st-gcn/resource/media/test.mp4 --openpose /home/chq/openpose/build
```
3. 如果你使用的是新版ST-GCN,请使用以下命令运行模型,并指定视频和Openpose路径:
```
python3 main.py demo --openpose '/home/chq/openpose/build' --video '/home/chq/st-gcn/resource/media/skateboarding.mp4'
```
请注意,以上引用提供了复现ST-GCN模型所需的安装和运行命令,你可以根据你的实际情况进行相应的调整和修改。
阅读全文