報錯如下：
starting epoch 1
/100.-
----
----
-traceback (most recent call last)
: file "e:/sharm zhao/3d_u-net/train.py"
, line 89,in
label_transforms=y_transforms)
file "e:/sharm zhao/3d_u-net/train.py"
, line 42
,in train
for img,label in dataloader:
file "c:\users\admin\anaconda3\envs\zhimingspace\lib\site-packages\torch\utils\data\dataloader.py"
, line 819
,in __next__
return self._process_data(data)
file "c:\users\admin\anaconda3\envs\zhimingspace\lib\site-packages\torch\utils\data\dataloader.py"
, line 846
,in _process_data
data.reraise(
) file "c:\users\admin\anaconda3\envs\zhimingspace\lib\site-packages\torch\_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:\users\admin\anaconda3\envs\zhimingspace\lib\site-packages\torch\utils\data\_utils\worker.py"
, line 178
,in _worker_loop
data = fetcher.fetch(index)
file "c:\users\admin\anaconda3\envs\zhimingspace\lib\site-packages\torch\utils\data\_utils\fetch.py"
, line 44
,in fetch
data =
[self.dataset[idx]
for idx in possibly_batched_index]
file "c:\users\admin\anaconda3\envs\zhimingspace\lib\site-packages\torch\utils\data\_utils\fetch.py"
, line 44,in
data =
[self.dataset[idx]
for idx in possibly_batched_index]
file "e:\sharm zhao\3d_u-net\dataset.py"
, line 38
,in __getitem__
img_y = sitk.getarrayfromimage(sitk.readimage(y_path)
).transpose((1
,2,0
))file "c:\users\admin\anaconda3\envs\zhimingspace\lib\site-packages\******itk\******itk.py"
, line 8876
,in readimage
return _******itk.readimage(
*args)
runtimeerror: exception thrown in ******itk readimage: d:\a\1\sitk\code\io\src\sitkimagereaderbase.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 ******itk readimage: d:\a\1\sitk\code\io\src\sitkimagereaderbase.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,傻了傻了！！！
PyTorch訓練集的讀取
pytorch讀取訓練資料是非常便捷的，只需要使用2個類 1 torch.utils.data.dataset 2 torch.utils.data.dataloader 常用資料集的讀取 1 torchvision.datasets的使用 對於常用資料集，可以使用torchvision.datas...
Pytorch中多GPU訓練
參考 在資料越來越多的時代，隨著模型規模引數的增多，以及資料量的不斷提公升，使用多gpu去訓練是不可避免的事情。pytorch在0.4.0及以後的版本中已經提供了多gpu訓練的方式，本文簡單講解下使用pytorch多gpu訓練的方式以及一些注意的地方。這裡我們談論的是單主機多gpus訓練，與分布式訓...
Pytorch 修改預訓練網路結構
我們以 inceptionv3 為例 pytorch裡我們如何使用設計好的網路結構，比如inceptionv3 import torchvision.models as models inception models.inception v3 pretrained true pytorch提供了個叫...