質問編集履歴

1

npyを作成したコードを追記しました。

2021/08/07 17:41

投稿

tkrd
tkrd

スコア5

test CHANGED
File without changes
test CHANGED
@@ -97,3 +97,75 @@
97
97
  ValueError: cannot reshape array of size 743050334 into shape (2768,512,1024)
98
98
 
99
99
  ```
100
+
101
+
102
+
103
+ 以下追記したソースコードですが、ここでは読み込んだ画像をndarrayに変換した後、512*1024のサイズになるようpadding(画像によって元々のサイズが違うため)した後、追加した軸に沿って画像2,768枚をstackしています。
104
+
105
+ ```python
106
+
107
+ import glob
108
+
109
+ import sunpy.map
110
+
111
+ import numpy as np
112
+
113
+ import matplotlib.pyplot as plt
114
+
115
+ import cv2
116
+
117
+ import math
118
+
119
+ from google.colab import drive
120
+
121
+ drive.mount('/content/drive', force_remount=True)
122
+
123
+
124
+
125
+ f_list = sorted(glob.glob('drive/My Drive/study/201005/*.magnetogram.fits'))
126
+
127
+ x_pad = 512
128
+
129
+ y_pad = 1024
130
+
131
+
132
+
133
+ input_data = np.array(sunpy.map.Map(f_list[0]).data)
134
+
135
+ pad_top = int((y_pad / 2) - math.floor(input_data.shape[1] / 2))
136
+
137
+ pad_bottom = int((y_pad / 2) - math.ceil(input_data.shape[1] / 2))
138
+
139
+ pad_left = int((x_pad / 2) - math.floor(input_data.shape[0] / 2))
140
+
141
+ pad_right = int((x_pad / 2) - math.ceil(input_data.shape[0] / 2))
142
+
143
+
144
+
145
+ input_data = np.pad(input_data, [(pad_left, pad_right), (pad_top, pad_bottom)], 'constant')
146
+
147
+ input_data = input_data[np.newaxis, :, :]
148
+
149
+ print(input_data.shape)
150
+
151
+ for i in range(len(f_list)-1):
152
+
153
+ tmp = np.array((sunpy.map.Map(f_list[i+1])).data)
154
+
155
+ pad_top = int((y_pad / 2) - math.floor(tmp.shape[1] / 2))
156
+
157
+ pad_bottom = int((y_pad / 2) - math.ceil(tmp.shape[1] / 2))
158
+
159
+ pad_left = int((x_pad / 2) - math.floor(tmp.shape[0] / 2))
160
+
161
+ pad_right = int((x_pad / 2) - math.ceil(tmp.shape[0] / 2))
162
+
163
+ tmp = np.pad(tmp, [(pad_left, pad_right), (pad_top, pad_bottom)], 'constant')
164
+
165
+ input_data = np.vstack([input_data, tmp[np.newaxis, :, :]])
166
+
167
+ print(input_data.shape)
168
+
169
+ np.save('drive/My Drive/study/temp_201005_All', input_data)
170
+
171
+ ```