test.py运行报错，预处理后的路径设置好了

[wangy-alt]

root@468c2519de72:/workspace/GaitPart# python train.py
Initialzing...
Initializing data source...
Loading training data...
libpng error: IDAT: incorrect data check
Traceback (most recent call last):
File "/opt/conda/lib/python3.8/site-packages/numpy/core/fromnumeric.py", line 56, in _wrapfunc
return getattr(obj, method)(*args, **kwds)
AttributeError: 'NoneType' object has no attribute 'reshape'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "train.py", line 18, in
m = initialization(conf, train=opt.cache)[0]
File "/workspace/GaitPart/model/initialization.py", line 57, in initialization
train_source, test_source = initialize_data(config, train, test)
File "/workspace/GaitPart/model/initialization.py", line 18, in initialize_data
train_source.load_all_data()
File "/workspace/GaitPart/model/utils/data_set.py", line 45, in load_all_data
self.load_data(i)
File "/workspace/GaitPart/model/utils/data_set.py", line 48, in load_data
return self.getitem(index)
File "/workspace/GaitPart/model/utils/data_set.py", line 62, in getitem
data = [self.loader(_path) for _path in self.seq_dir[index]]
File "/workspace/GaitPart/model/utils/data_set.py", line 62, in
data = [self.loader(_path) for _path in self.seq_dir[index]]
File "/workspace/GaitPart/model/utils/data_set.py", line 51, in loader
return self.img2xarray(
File "/workspace/GaitPart/model/utils/data_set.py", line 76, in img2xarray
frame_list = [np.reshape(
File "/workspace/GaitPart/model/utils/data_set.py", line 76, in
frame_list = [np.reshape(
File "/opt/conda/lib/python3.8/site-packages/numpy/core/fromnumeric.py", line 292, in reshape
return _wrapfunc(a, 'reshape', newshape, order=order)
File "/opt/conda/lib/python3.8/site-packages/numpy/core/fromnumeric.py", line 66, in _wrapfunc
return _wrapit(obj, method, *args, **kwds)
File "/opt/conda/lib/python3.8/site-packages/numpy/core/fromnumeric.py", line 46, in _wrapit
result = getattr(asarray(obj), method)(*args, **kwds)
ValueError: cannot reshape array of size 1 into shape (64,64,newaxis)

[wyp196]

我也是这个问题，请问您解决了吗

[zhang123-sys]

raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for SetNet:
Missing key(s) in state_dict: "set_layer1.forward_block.conv.weight", "set_layer2.forward_block.conv.weight", "set_layer3.forward_block.conv.weight", "set_layer4.forward_block.conv.weight", "set_layer5.forward_block.conv.weight", "set_layer6.forward_block.conv.weight", "gl_layer1.conv.weight", "gl_layer2.conv.weight", "gl_layer3.conv.weight", "gl_layer4.conv.weight", "fc_bin.0".
Unexpected key(s) in state_dict: "module.set_layer1.forward_block.conv.weight", "module.set_layer2.forward_block.conv.weight", "module.set_layer3.forward_block.conv.weight", "module.set_layer4.forward_block.conv.weight", "module.set_layer5.forward_block.conv.weight", "module.set_layer6.forward_block.conv.weight", "module.gl_layer1.conv.weight", "module.gl_layer2.conv.weight", "module.gl_layer3.conv.weight", "module.gl_layer4.conv.weight", "module.fc_bin.0".