关于CTR-GCN源码复现相关问题

运行环境Anaconda23.1.0、 pytorch1.13.1、 python3.9、 pycharm 按照README文件安装相关库(因为部分包的版本问题直接pip install -r requirements.txt中途会报错,修改相关包版本如下) 一、将UVA-Human.npz改名NTU60_CS.npz放在CTR-GCN-main/data/ntu 文件夹下(这样不用下载数据集自己处理生成,关于数据集预处理这里不赘述了),如下图所示: 二、Main.py的修改 main.py中的下面代码我给注释了,因为在windows环境下,在Linux环境下可以忽略。

import resource

rlimit = resource.getrlimit(resource.RLIMIT_NOFILE)

resource.setrlimit(resource.RLIMIT_NOFILE, (2048, rlimit[1]))

564行代码修改如下(报Loader error)

default_arg = yaml.safe_load(f.read())

三、配置文件config修改 CTR-GCN-main/config/nturgbd-cross-subject中 num_class: 10 num_point: 17(UVA-Human数据集中骨架节点17) 四、feeders修改 CTR-GCN-main/feeders/feeder_ntu修改 节点数25->17

self.data = self.data.reshape((N, T, 2, 17, 3)).transpose(0, 4, 1, 3, 2)

报以下两个包错误(no module DictAction/torchpack.runner)保证pip install -e torchlight之后,我现在的torchpack==0.0.3版本安装自己去百度搜torchpack当前版本安装,直接pip install torchpack版本可能不对就会报这种错误。

from torchpack.runner.hooks import PaviLogger

from torchlight import DictAction

相关链接

评论可见,请评论后查看内容,谢谢!!!
 您阅读本篇文章共花了: