小样本金属缺陷分割复现代码报错。

Traceback (most recent call last): File "E:/postgraduate_paper/WZ/CPANet/CPANet/train.py", line 466, in <module> main() File "E:/postgraduate_paper/WZ/CPANet/CPANet/train.py", line 71, in main main_worker(args) File "E:/postgraduate_paper/WZ/CPANet/CPANet/train.py", line 218, in main_worker loss_train, mIoU_train, mAcc_train, allAcc_train = train(train_loader, model, optimizer, epoch) File "E:/postgraduate_paper/WZ/CPANet/CPANet/train.py", line 271, in train for i, (input, target, s_input, s_mask, subcls, _) in enumerate(train_loader): File "D:\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 521, in __next__ data = self._next_data() File "D:\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 1203, in _next_data return self._process_data(data) File "D:\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 1229, in _process_data data.reraise() File "D:\Anaconda\envs\pytorch\lib\site-packages\torch\_utils.py", line 434, in reraise raise exception cv2.error: Caught error in DataLoader worker process 0. Original Traceback (most recent call last): File "D:\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\worker.py", line 287, in _worker_loop data = fetcher.fetch(index) File "D:\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\fetch.py", line 49, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "D:\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\fetch.py", line 49, in <listcomp> data = [self.dataset[idx] for idx in possibly_batched_index] File "E:\postgraduate_paper\WZ\CPANet\CPANet\util\dataset.py", line 85, in __getitem__ query_rgb = cv2.cvtColor(query_rgb, cv2.COLOR_BGR2RGB) cv2.error: OpenCV(3.4.2) c:\miniconda3\conda-bld\opencv-suite_1534379934306\work\modules\imgproc\src\color.hpp:253: error: (-215:Assertion failed) VScn::contains(scn) && VDcn::contains(dcn) && VDepth::contains(depth) in function 'cv::CvtHelper<struct cv::Set<3,4,-1>,struct cv::Set<3,4,-1>,struct cv::Set<0,2,5>,2>::CvtHelper' Process finished with exit code 1

有没有相似错误的大佬，麻烦指导，感激不尽。