deep-learning-for-image-processing icon indicating copy to clipboard operation
deep-learning-for-image-processing copied to clipboard

TypeError: Argument 'bb' has incorrect type (expected numpy.ndarray, got list)

Open HRruiH opened this issue 9 months ago • 0 comments

运行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)

HRruiH avatar Mar 01 '25 15:03 HRruiH