Advertisement

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)

还没有任何评论哟~