pytorch在网络训练中,读取数据时出现问题
报错如下:
Starting epoch 1/100.
----------
Traceback (most recent call last):
File "E:/Sharm Zhao/3D_U-net/train.py", line 89, in <module>
label_transforms=y_transforms)
File "E:/Sharm Zhao/3D_U-net/train.py", line 42, in train
for img,label in dataloader:
File "C:UsersadminAnaconda3envszhimingspacelibsite-packages orchutilsdatadataloader.py", line 819, in __next__
return self._process_data(data)
File "C:UsersadminAnaconda3envszhimingspacelibsite-packages orchutilsdatadataloader.py", line 846, in _process_data
data.reraise()
File "C:UsersadminAnaconda3envszhimingspacelibsite-packages orch\_utils.py", line 369, in reraise
raise self.exc_type(msg)
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:UsersadminAnaconda3envszhimingspacelibsite-packages orchutilsdata\_utilsworker.py", line 178, in _worker_loop
data = fetcher.fetch(index)
File "C:UsersadminAnaconda3envszhimingspacelibsite-packages orchutilsdata\_utilsfetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:UsersadminAnaconda3envszhimingspacelibsite-packages orchutilsdata\_utilsfetch.py", line 44, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "E:Sharm Zhao3D_U-netdataset.py", line 38, in __getitem__
img_y = sitk.GetArrayFromImage(sitk.ReadImage(y_path)).transpose((1,2,0))
File "C:UsersadminAnaconda3envszhimingspacelibsite-packagesSimpleITKSimpleITK.py", line 8876, in ReadImage
return _SimpleITK.ReadImage(*args)
RuntimeError: Exception thrown in SimpleITK ReadImage: D:a1sitkCodeIOsrcsitkImageReaderBase.cxx:99:
sitk::ERROR: The file "E:/Sharm Zhao/3D_U-net/data/Task02_Heart/after_slice/labelsTr/seg-3043.nii" does not exist.
有几个关键的问题: 1、RuntimeError: Caught RuntimeError in DataLoader worker process 0 2、RuntimeError: Exception thrown in SimpleITK ReadImage: D:a1sitkCodeIOsrcsitkImageReaderBase.cxx:99: 3、sitk::ERROR: The file “E:/Sharm Zhao/3D_U-net/data/Task02_Heart/after_slice/labelsTr/seg-3043.nii” does not exist.
在这个问题上找了好久,一直不知道到底是哪里出错了。。。
最后才发现,我的标签的文件夹的名称是 labelsTs,而不是labelsTr,傻了傻了!!!
上一篇:
JS实现多线程数据分片下载
