search

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

ZeroDivisionError while running test.py #83

Open keshav-qubitrics opened 3 years ago

keshav-qubitrics commented 3 years ago

Hi everyone,

I am facing ZeroDivisionError while running test.py. Here is the error:

2021-03-01 06:43:06.797786: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.10.1 Loading checkpoint: /content/drive/MyDrive/Pick.Pytorch.Sroie/saved/models/PICK_Default/test_0227_063059/model_best.pth with saved mEF 0.8032 ... boxes_and_transcripts_folder /content/drive/MyDrive/test_boxes_and_transcripts 100it [02:42, 1.62s/it] Traceback (most recent call last): File "/content/drive/MyDrive/Pick.Pytorch.Sroie/test.py", line 104, in main(args) File "/content/drive/MyDrive/Pick.Pytorch.Sroie/test.py", line 50, in main for step_idx, input_data_item in tqdm(enumerate(test_data_loader)): File "/usr/local/lib/python3.7/dist-packages/tqdm/std.py", line 1129, in iter for obj in iterable: File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 345, in next data = self._next_data() File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 838, in _next_data return self._process_data(data) File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 881, in _process_data data.reraise() File "/usr/local/lib/python3.7/dist-packages/torch/_utils.py", line 395, in reraise raise self.exc_type(msg) RuntimeError: Caught RuntimeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/content/drive/MyDrive/Pick.Pytorch.Sroie/data_utils/documents.py", line 119, in init transcripts) File "/content/drive/MyDrive/Pick.Pytorch.Sroie/data_utils/documents.py", line 202, in relation_features_between_ij_nodes if width_i / (height_i) is not None else -1 # w_i/h_i ZeroDivisionError: float division by zero

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "/content/drive/MyDrive/Pick.Pytorch.Sroie/data_utils/pick_dataset.py", line 115, in getitem image_index=index, training=self.training) File "/content/drive/MyDrive/Pick.Pytorch.Sroie/data_utils/documents.py", line 169, in init raise RuntimeError('Error occurs in image {}: {}'.format(boxes_and_transcripts_file.stem, e.args)) RuntimeError: Error occurs in image X51006008092: ('float division by zero',)

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop data = fetcher.fetch(index) File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/content/drive/MyDrive/Pick.Pytorch.Sroie/data_utils/pick_dataset.py", line 123, in getitem raise RuntimeError('Error occurs in image {}: {}'.format(boxes_and_transcripts_file.stem, e.args)) RuntimeError: Error occurs in image X51006008092: ("Error occurs in image X51006008092: ('float division by zero',)",)