def test_conv_layer_idxs(self):
network = Network(
model_fn='../ggcnn/data/networks/ggcnn_rss/epoch_29_model.hdf5')
layers = network.conv_layer_idxs
self.assertTrue([1, 2, 3] == layers)
deeper = network.deeper(3)
layers2 = deeper.conv_layer_idxs
self.assertTrue(layers2 == [1, 2, 3, 4])
def test_get_expandable_layer_idxs(self):
network = Network(
model_fn='../ggcnn/data/networks/ggcnn_rss/epoch_29_model.hdf5')
layers = network.get_expandable_layer_idxs()
self.assertTrue([1, 2, 3] == layers)
deeper = network.deeper(3)
layers2 = deeper.conv_layer_idxs
self.assertTrue(layers2 == [1, 2, 3, 4])
layers = network.get_expandable_layer_idxs(transpose=True)
self.assertTrue([1, 2, 3, 4, 5, 6] == layers)
def test_deeper(self):
network = Network(
model_fn='../ggcnn/data/networks/ggcnn_rss/epoch_29_model.hdf5')
network2 = network.deeper(layer=2)
network3 = network.deeper(layer=5)
network4 = network.deeper(layer=6)
input_img = np.expand_dims(np.load('depth_inpainted.npy'), axis=2)
input_img = np.expand_dims(input_img, axis=0)
output1 = network.predict(input_img)
output2 = network2.predict(input_img)
output3 = network3.predict(input_img)
output4 = network4.predict(input_img)
self.assertTrue(
str(network2) ==
'C9x9x32_C5x5x16_C5x5x16_C3x3x8_T3x3x8_T5x5x16_T9x9x32')
self.assertTrue(
str(network3) ==
'C9x9x32_C5x5x16_C3x3x8_T3x3x8_T5x5x16_T5x5x16_T9x9x32')
self.assertTrue(
str(network4) ==
'C9x9x32_C5x5x16_C3x3x8_T3x3x8_T5x5x16_T9x9x32_T9x9x32')
for o in [output2, output3, output4]:
self.assertTrue((o[0] == output1[0]).all())
