質問編集履歴
1
コードの変更
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -16,59 +16,8 @@
|
|
|
16
16
|
### 該当のソースコード
|
|
17
17
|
|
|
18
18
|
```python
|
|
19
|
-
#%%
|
|
20
|
-
#mhd読み込みtrain
|
|
21
|
-
def load_itk(filename):
|
|
22
|
-
imglist = []
|
|
23
|
-
spacinglist = []
|
|
24
|
-
|
|
25
|
-
for i in range(len(train_a_FILE)):
|
|
26
|
-
itkimage = sitk.ReadImage(filename)
|
|
27
|
-
raw_image = sitk.GetArrayFromImage(itkimage)
|
|
28
|
-
raw_array = np.asarray(raw_image, dtype=np.float32)
|
|
29
|
-
# raw_array = raw_array.transpose(1, 2, 0)
|
|
30
|
-
|
|
31
|
-
imglist.append(raw_array)
|
|
32
|
-
|
|
33
|
-
raw_spacing = itkimage.GetSpacing()
|
|
34
|
-
spacinglist.append(raw_spacing)
|
|
35
|
-
imgdatas = np.asarray(imglist)
|
|
36
|
-
return raw_array , imgdatas , spacinglist
|
|
37
19
|
|
|
38
20
|
|
|
39
|
-
#%%
|
|
40
|
-
###train
|
|
41
|
-
train_a_datalist = []
|
|
42
|
-
train_b_datalist=[]
|
|
43
|
-
for i in range(len(train_FILE)):
|
|
44
|
-
a_train_image,aimagedatas,a_train_spacinglist=load_itk(train_a_FILE[i])
|
|
45
|
-
b_train_image,bimagedatas,b_train_spacinglist=load_itk(train_b_FILE[i])
|
|
46
|
-
|
|
47
|
-
#入力チェック時のエラー：input_1が4次元であることを期待したが、形状が(48, 180, 180)の配列を得たとエラーが出るので、newaxisで次元追加
|
|
48
|
-
a_train_image=a_train_image[:,:,:,np.newaxis]
|
|
49
|
-
a_train_image=a_train_image[:,:,:,np.newaxis]
|
|
50
|
-
a_train_image = np.array(a_train_image, dtype=np.float32)
|
|
51
|
-
b_train_image = np.array(b_train_image, dtype=np.float32)
|
|
52
|
-
|
|
53
|
-
#画像の4分割したものをとる
|
|
54
|
-
a_len = len(a_train_image)
|
|
55
|
-
a_min_i = a_len//4
|
|
56
|
-
a_max_i= a_len - a_min_i
|
|
57
|
-
|
|
58
|
-
b_len = len(b_train_image)
|
|
59
|
-
b_min_i = b_len//4
|
|
60
|
-
b_max_i= b_len - b_min_i
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
train_a_slice=np.array(a_train_image[a_min_i:a_max_i,:,:])
|
|
64
|
-
|
|
65
|
-
train_a_datalist.append(train_a_slice)
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
train_b_slice=np.array(b_train_image[b_min_i:b_max_i,:,:])
|
|
69
|
-
|
|
70
|
-
train_b_datalist.append(train_b_slice)
|
|
71
|
-
|
|
72
21
|
#スライス画像をゼロパディングする→元画像のshape(49, 180, 180, 1)から(49, 182, 182, 1)に変更されていることを確認
|
|
73
22
|
train_a_slisepadding=np.pad(train_a_slice,((0,0),(1,1),(1,1),(0,0)),'constant')
|
|
74
23
|
train_b_slisepadding=np.pad(train_b_slice,((0,0),(1,1),(1,1),(0,0)),'constant')
|