patch_w=patch_w,
patch_h=patch_h,
patch_depth=n_channels)
sess = tf.InteractiveSession()
with sess.as_default():
print('Loading model')
model = unet(inputshape=(patch_w, patch_h, n_channels),
conv_depth=conv_depth)
model.load_weights(args.weights)
print('Loading charge info')
if steps == 0:
test_charge = np.zeros(
(test_gen.__len__() * batch_size, patch_w, patch_h, 1))
test_energy = np.zeros(
(test_gen.__len__() * batch_size, patch_w, patch_h, 1))
for i in range(test_gen.__len__()):
wires = test_gen.getitembykey(i, 'wire')
energies = test_gen.getitembykey(i, 'energy')
for j in range(batch_size):
test_charge[(i * batch_size) + j] = wires[j]
test_energy[(i * batch_size) + j] = energies[j]
else:
test_charge = np.zeros((steps * batch_size, patch_w, patch_h, 1))
test_energy = np.zeros((steps * batch_size, patch_w, patch_h, 1))
for i in range(steps):
wires = test_gen.getitembykey(i, 'wire')
energies = test_gen.getitembykey(i, 'energy')
for j in range(batch_size):
shuffle=False,
root_data=args.input,
patch_w=patch_w,
patch_h=patch_h,
patch_depth=n_channels)
sess = tf.InteractiveSession()
with sess.as_default():
print('Loading model')
model = unet(inputshape=(patch_w, patch_h, n_channels),
conv_depth=conv_depth)
model.load_weights(args.weights)
print('Reformating data')
test_x = np.zeros((test_gen.__len__(), patch_w, patch_h, n_channels))
test_y = np.zeros((test_gen.__len__(), patch_w, patch_h, 1))
test_charge = np.zeros((test_gen.__len__(), patch_w, patch_h, 1))
test_energy = np.zeros((test_gen.__len__(), patch_w, patch_h, 1))
for i in range(test_gen.__len__()):
test_x[i], test_y[i] = test_gen.__getitem__(i)
test_charge[i] = test_gen.getitembykey(i, 'wire')
test_energy[i] = test_gen.getitembykey(i, 'energy')
# FIXME
# test_x = test_x[:8]
# test_y = test_y[:8]
# test_charge = test_charge[:8]
# test_energy = test_energy[:8]
print('Making predictions')
