Traceback (most recent call last):


 "test.py", line 285, in <module>

main()


 "test.py", line 267, in main

optimizer, device)


 "test.py", line 130, in train

for i, (pos_data, neg_data) in enumerate(zip(pos_loader, neg_loader)):


 "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 345, in __next__

data = self._next_data()


 "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 856, in _next_data

return self._process_data(data)


 "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 881, in _process_data

data.reraise()


 "/usr/local/lib/python3.6/dist-packages/torch/_utils.py", line 394, in reraise

raise self.exc_type(msg)
AttributeError: Caught AttributeError in DataLoader worker process 0.

Traceback (most recent call last):


 "train.py", line 43, in prefetch_data

data, ind = sample_data(db, ind, data_aug=data_aug)


 "/home/xinyu/CenterNet-master/sample/coco.py", line 199, in sample_data

return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)


 "/home/xinyu/CenterNet-master/sample/coco.py", line 99, in kp_detection

image, detections = random_crop(image, detections, rand_scales, input_size, border=border)


 "/home/xinyu/CenterNet-master/sample/utils.py", line 57, in random_crop

image_height, image_width = image.shape[0:2]
AttributeError: 'NoneType' object has no attribute 'shape'
