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())