def test_hourglass_52(self): net = hourglass.hourglass_52(2, downsample=False) self.assertEqual(hourglass.hourglass_depth(net), 52) outputs = net(tf.zeros((2, 32, 32, 3))) self.assertEqual(outputs[0].shape, (2, 32, 32, 4))
def test_hourglass_104(self): net = hourglass.hourglass_104() self.assertEqual(hourglass.hourglass_depth(net), 104)
def test_hourglass_100(self): net = hourglass.hourglass_100(2, initial_downsample=False) self.assertEqual(hourglass.hourglass_depth(net), 100) outputs = net(tf.zeros((2, 32, 32, 3))) self.assertEqual(outputs[0].shape, (2, 32, 32, 4))
def test_hourglass_20_uniform_size(self): net = hourglass.hourglass_20_uniform_size(2) self.assertEqual(hourglass.hourglass_depth(net), 20) outputs = net(tf.zeros((2, 32, 32, 3))) self.assertEqual(outputs[0].shape, (2, 32, 32, 4))
def test_hourglass_52(self): net = hourglass.hourglass_52(2) self.assertEqual(hourglass.hourglass_depth(net), 52)
def test_hourglass_20(self): net = hourglass.hourglass_20(2) self.assertEqual(hourglass.hourglass_depth(net), 20)