Description
运行mask-rcnn，第一次test时报这个错误
Traceback (most recent call last):
File "train.py", line 240, in
main(args)
File "train.py", line 149, in main
det_info, seg_info = utils.evaluate(model, val_data_loader, device=device)
File "/opt/conda/lib/python3.8/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/mnt/workspace/mask_rcnn/train_utils/train_eval_utils.py", line 77, in evaluate
for image, targets in metric_logger.log_every(data_loader, 100, header):
File "/mnt/workspace/mask_rcnn/train_utils/distributed_utils.py", line 180, in log_every
for obj in iterable:
File "/opt/conda/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 633, in next
data = self._next_data()
File "/opt/conda/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1345, in _next_data
return self._process_data(data)
File "/opt/conda/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1371, in _process_data
data.reraise()
File "/opt/conda/lib/python3.8/site-packages/torch/_utils.py", line 644, in reraise
raise exception
TypeError: Caught TypeError in DataLoader worker process 1.
Original Traceback (most recent call last):
File "/opt/conda/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
data = fetcher.fetch(index)
File "/opt/conda/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/opt/conda/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/mnt/workspace/mask_rcnn/my_dataset_coco.py", line 128, in getitem
target = self.parse_targets(img_id, coco_target, w, h)
File "/mnt/workspace/mask_rcnn/my_dataset_coco.py", line 89, in parse_targets
masks = convert_coco_poly_mask(segmentations, h, w)
File "/mnt/workspace/mask_rcnn/train_utils/coco_utils.py", line 43, in convert_coco_poly_mask
rles = coco_mask.frPyObjects(polygons, height, width)
File "pycocotools/_mask.pyx", line 294, in pycocotools._mask.frPyObjects
TypeError: Argument 'bb' has incorrect type (expected numpy.ndarray, got list)