关于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
相关链接
发表评论