Error when training on other dataset

[CBS5181]

Hi, author. Thanks for your nice work! I tried to train on the ZInD dataset, but I got some error message. I don't change any code you provided. By the way, should I use data augmentation if I have 20077 train set size?

• Environment Ubuntu 20.04.5 LTS

• Command

python train.py --cfg_file cfgs/zind.yaml --id zind -b 8 --train_root /home/user/CBS/DMH-Net/data/zind_layout/train --valid_root_dir /home/user/CBS/DMH-Net/data/zind_layout/valid

• Error Message

num_workers: 8
batch_size: 8
train_set_size: 20077
LR 0.000100
Train ep1:   0%|                                       | 0/2510 [00:00<?, ?it/s]
Epoch:   0%|                                             | 0/75 [00:00<?, ?ep/s]
Traceback (most recent call last):
  File "train.py", line 320, in <module>
    input = next(iterator_train)
  File "/home/user/anaconda3/envs/dmhnet/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 521, in __next__
    data = self._next_data()
  File "/home/user/anaconda3/envs/dmhnet/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1203, in _next_data
    return self._process_data(data)
  File "/home/user/anaconda3/envs/dmhnet/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1229, in _process_data
    data.reraise()
  File "/home/user/anaconda3/envs/dmhnet/lib/python3.7/site-packages/torch/_utils.py", line 425, in reraise
    raise self.exc_type(msg)
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/home/user/anaconda3/envs/dmhnet/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
    data = fetcher.fetch(index)
  File "/home/user/anaconda3/envs/dmhnet/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/user/anaconda3/envs/dmhnet/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/user/Laurence/DMH-Net/perspective_dataset.py", line 101, in __getitem__
    return self.getItem(self.img_fnames[idx], self.cors[idx])
  File "/home/user/Laurence/DMH-Net/perspective_dataset.py", line 286, in getItem
    for line in d["lines"]:
KeyError: 'lines'