FilesExpand file tree

master

/

input_data.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

25 lines (24 loc) · 893 Bytes

master

/

input_data.py

Copy path

Top

File metadata and controls

• Code
• Blame

25 lines (24 loc) · 893 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

import scipy.io as sio

#import h5py

data = sio.loadmat('./data/64to32_11sources_middle1999_ellipse_value_6.mat')

#data = h5py.File('./data/test_64_circle_1e4.mat')

Sig_set_reshape = data["Sig_set_reshape"]

dis_set_reshape = data["dis_set_reshape"]

N_test = data["N_test"]

U_ob_reshape = data["U_ob_reshape"]

def input_data(test):

if test:

x = Sig_set_reshape[int(N_test[0][0] * 0.8):]

x = x.reshape(-1,64,64,1)

d = dis_set_reshape[int(N_test[0][0] * 0.8):]

d = d.reshape(-1,64,64,1)

y = U_ob_reshape[int(N_test[0][0] * 0.8):]

y = y.reshape(-1,32,32,1)

else:

x = Sig_set_reshape[:int(N_test[0][0] * 0.8)]

x = x.reshape(-1,64,64,1)

d = dis_set_reshape[:int(N_test[0][0] * 0.8)]

d = d.reshape(-1,64,64,1)

y = U_ob_reshape[:int(N_test[0][0] * 0.8)]

y = y.reshape(-1,32,32,1)

return x,d,y