-
Notifications
You must be signed in to change notification settings - Fork 130
Description
These errors look like there is a problem with bgr2hsv. There is a problem with the channel. I don’t know how it happened.
Traceback (most recent call last):
cv2.error: Caught error in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/root/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/root/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/autodl-tmp/SurroundOcc/SurroundOcc-main/projects/mmdet3d_plugin/datasets/nuscenes_occupancy_dataset.py", line 132, in getitem
data = self.prepare_train_data(idx)
File "/root/autodl-tmp/SurroundOcc/SurroundOcc-main/projects/mmdet3d_plugin/datasets/nuscenes_occupancy_dataset.py", line 47, in prepare_train_data
example = self.pipeline(input_dict)
File "/root/miniconda3/envs/surroundocc/lib/python3.7/site-packages/mmdet/datasets/pipelines/compose.py", line 40, in call
data = t(data)
File "/root/autodl-tmp/SurroundOcc/SurroundOcc-main/projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py", line 158, in call
img = mmcv.bgr2hsv(img)
File "/root/miniconda3/envs/surroundocc/lib/python3.7/site-packages/mmcv/image/colorspace.py", line 280, in convert_color
out_img = cv2.cvtColor(img, code)
cv2.error: OpenCV(4.5.5) /io/opencv/modules/imgproc/src/color.simd_helpers.hpp:92: error: (-2:Unspecified error) in function 'cv::impl::{anonymous}::CvtHelper<VScn, VDcn, VDepth, sizePolicy>::CvtHelper(cv::InputArray, cv::OutputArray, int) [with VScn = cv::impl::{anonymous}::Set<3, 4>; VDcn = cv::impl::{anonymous}::Set<3>; VDepth = cv::impl::{anonymous}::Set<0, 5>; cv::impl::{anonymous}::SizePolicy sizePolicy = cv::impl::::NONE; cv::InputArray = const cv::_InputArray&; cv::OutputArray = const cv::_OutputArray&]'
Invalid number of channels in input image:
'VScn::contains(scn)'
where
'scn' is 1