pytorch mxnet ValueError: too many dimensions ‘NDArray‘
发布时间
阅读量:
阅读量
Traceback (most recent call last):
File "topFAR_COX_py2_fc1_pytorch_PY3.py", line 199, in <module>
IDimage_features_dict = getfeatures_dict(model, IDimage_list, IDimage_path, featurelen)
File "topFAR_COX_py2_fc1_pytorch_PY3.py", line 106, in getfeatures_dict
emb = extractfeature(model, oriimg)
File "/home/user1/miniconda3/envs/py377/lib/python3.7/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "topFAR_COX_py2_fc1_pytorch_PY3.py", line 60, in extractfeature
img = torch.Tensor(img).cuda()
全部评论 (0)
还没有任何评论哟~
