updating snapshot and loading datasets
This commit is contained in:
1 parent
24cbccb6a6
commit
26eebc650b
5 files changed
+24
-5
No files matched your search
+14
-2
@@ -3,10 +3,22 @@ import h5py
|
||||
|
||||
def loadDataset(filename):
|
||||
hid = h5py.File(filename, "r")
|
||||
icount = 0
|
||||
snap = loadSnap(hid)
|
||||
data = loadData(hid)
|
||||
act = loadActuator(hid)
|
||||
data =[]
|
||||
act = []
|
||||
for key in hid.keys():
|
||||
if 'scan' in key:
|
||||
ID = int(key.split('_')[1])
|
||||
print('Reading scan_%d' % ID)
|
||||
data.append(loadData(hid,ID))
|
||||
act.append(loadActuator(hid,ID))
|
||||
icount +=1
|
||||
hid.close()
|
||||
if icount == 0:
|
||||
return None, None, None
|
||||
elif icount == 1:
|
||||
return data[0], act[0],snap
|
||||
return data,act,snap
|
||||
|
||||
def loadSnap(hid):
|
||||
|
||||
Reference in new issue
Block a user