有人跑过ST-GCN吗

2021/4/10镜像同步6 回复

求问跑了https://github.com/yysijie/st-gcn原作者的代码在train的时候有这个问题RuntimeError: Given groups=1, weight of size [60, 256, 1, 1], expected input[16, 3, 1, 1] to have 256 channels, but got 3 channels instead 是代码有错误吧

订阅后，新回复会通过你的通知中心匿名送达。

6 条回复

SuHollow机器人#1 · 2021/4/10

通道数量错误吧，预期输入是256通道，但是实际你的数据只有3通道，建议看一下是不是输入数据预处理出问题了

yt1997机器人#2 · 2021/4/10

可以看看图的邻接矩阵对不对

hhhytycwfp机器人#3 · 2021/4/10

是的谢谢哟【在 SuHollow (pikdjf) 的大作中提到: 】 : 通道数量错误吧，预期输入是256通道，但是实际你的数据只有3通道，建议看一下是不是输入数据预处理出问题了

hhhytycwfp机器人#4 · 2021/4/10

嗯嗯ok 谢谢【在 yt1997 (john_snow) 的大作中提到: 】 : 可以看看图的邻接矩阵对不对

GoogolX机器人#5 · 2021/4/10

没跑过这个模型，不知道数据维度是啥样的。我就看了下st_gcn.py，应该就是第89行的x = self.fcn(x)这个地方的通道数出了问题，维度不对。有俩个可能，一个是他模型跑的数据跟你跑的数据不一样，一个是他代码可能有点小问题。这个只能你多检查检查数据维度，或者print出来看数据维度是哪出了问题。

hhhytycwfp机器人#6 · 2021/4/10

谢谢你！确实是传给fcn的维度有问题数据和我的一样用的他的数据集预处理这篇文章用了两个数据集我用了其中一个网上也有人说用ntu-gbd数据集时有维度不对的问题我怕改了就不是人家的网络了?差一点结果可能都不一样【在 GoogolX (GoogolX) 的大作中提到: 】 : 没跑过这个模型，不知道数据维度是啥样的。我就看了下st_gcn.py，应该就是第89行的x = self.fcn(x)这个地方的通道数出了问题，维度不对。有俩个...