wenwenyu / PICK-pytorch

Code for the paper "PICK: Processing Key Information Extraction from Documents using Improved Graph Learning-Convolutional Networks" (ICPR 2020)
https://arxiv.org/abs/2004.07464
MIT License
553 stars 191 forks source link

Problem during test #98

Open mariotiziomario opened 2 years ago

mariotiziomario commented 2 years ago

It seems that there is some problem with normalize_relation_features, File "/mnt/batch/tasks/shared/LS_root/mounts/clusters/test-bello/code/Users/t/PICK-pytorch/data_utils/documents.py", line 136, in __init__ relation_features = normalize_relation_features(relation_features, width=width, height=height) File "/mnt/batch/tasks/shared/LS_root/mounts/clusters/test-bello/code/Users/t/PICK-pytorch/data_utils/documents.py", line 299, in normalize_relation_features max_value = np.max(feat_ij) File "/anaconda/envs/azureml_py38/lib/python3.8/site-packages/numpy/core/fromnumeric.py", line 2504, in amax return _wrapreduction(a, np.maximum, 'max', axis, None, out, keepdims=keepdims, File "/anaconda/envs/azureml_py38/lib/python3.8/site-packages/numpy/core/fromnumeric.py", line 86, in _wrapreduction return ufunc.reduce(obj, axis, dtype, out, **passkwargs) ValueError: zero-size array to reduction operation maximum which has no identity, can you help me to solve this? Thank you