djiajunustc/Voxel-R-CNN

Pickle file errors

Closed this issue · 1 comments

Sidu28 commented

On Ubunut 18.04. I am a bit unsure as to why this might be happening-it happens when I run the train shell script:

Traceback (most recent call last): File "<string>", line 1, in <module> File "/usr/lib/python3.8/multiprocessing/spawn.py", line 116, in spawn_main exitcode = _main(fd, parent_sentinel) File "/usr/lib/python3.8/multiprocessing/spawn.py", line 126, in _main self = reduction.pickle.load(from_parent) _pickle.UnpicklingError: pickle data was truncated Traceback (most recent call last): File "<string>", line 1, in <module> File "/usr/lib/python3.8/multiprocessing/spawn.py", line 116, in spawn_main exitcode = _main(fd, parent_sentinel) File "/usr/lib/python3.8/multiprocessing/spawn.py", line 126, in _main self = reduction.pickle.load(from_parent) _pickle.UnpicklingError: pickle data was truncated

After this error shows up, I get a key error for road planes:
Original Traceback (most recent call last): File "/usr/local/lib/python3.8/dist-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop data = fetcher.fetch(index) File "/usr/local/lib/python3.8/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.8/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp> data = [self.dataset[idx] for idx in possibly_batched_index] File "/data/Voxel-R-CNN/pcdet/datasets/kitti/kitti_dataset.py", line 380, in __getitem__ data_dict = self.prepare_data(data_dict=input_dict) File "/data/Voxel-R-CNN/pcdet/datasets/dataset.py", line 121, in prepare_data data_dict = self.data_augmentor.forward( File "/data/Voxel-R-CNN/pcdet/datasets/augmentor/data_augmentor.py", line 94, in forward data_dict = cur_augmentor(data_dict=data_dict) File "/data/Voxel-R-CNN/pcdet/datasets/augmentor/database_sampler.py", line 197, in __call__ data_dict = self.add_sampled_boxes_to_scene(data_dict, sampled_gt_boxes, total_valid_sampled_dict) File "/data/Voxel-R-CNN/pcdet/datasets/augmentor/database_sampler.py", line 125, in add_sampled_boxes_to_scene sampled_gt_boxes, data_dict['road_plane'], data_dict['calib'] KeyError: 'road_plane'