dbolya / yolact

A simple, fully convolutional model for real-time instance segmentation.
MIT License
5k stars 1.33k forks source link

"mtrand.pyx", line 936, in numpy.random.mtrand.RandomState.choice #843

Closed sabdha closed 7 months ago

sabdha commented 7 months ago

I am facing the below error when I start training. There was no issue before. Begin training!

Traceback (most recent call last): File "/content/drive/MyDrive/yolact/./train.py", line 504, in train() File "/content/drive/MyDrive/yolact/./train.py", line 270, in train for datum in data_loader: File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py", line 630, in next data = self._next_data() File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py", line 1345, in _next_data return self._process_data(data) File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py", line 1371, in _process_data data.reraise() File "/usr/local/lib/python3.10/dist-packages/torch/_utils.py", line 694, in reraise raise exception ValueError: Caught ValueError in DataLoader worker process 0. Original Traceback (most recent call last): File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/content/drive/MyDrive/yolact/data/coco.py", line 94, in getitem im, gt, masks, h, w, num_crowds = self.pull_item(index) File "/content/drive/MyDrive/yolact/data/coco.py", line 158, in pull_item img, masks, boxes, labels = self.transform(img, masks, target[:, :4], File "/content/drive/MyDrive/yolact/utils/augmentations.py", line 688, in call return self.augment(img, masks, boxes, labels) File "/content/drive/MyDrive/yolact/utils/augmentations.py", line 55, in call img, masks, boxes, labels = t(img, masks, boxes, labels) File "/content/drive/MyDrive/yolact/utils/augmentations.py", line 309, in call mode = random.choice(self.sample_options) File "mtrand.pyx", line 936, in numpy.random.mtrand.RandomState.choice ValueError: setting an array element with a sequence. The requested array has an inhomogeneous shape after 1 dimensions. The detected shape was (6,) + inhomogeneous part.

sabdha commented 7 months ago

This issue got resolved when I downgraged numpy to 1.23.2.

pipi0816 commented 6 months ago

thanks