Python hourglass_depth Examples

Example #1

File: hourglass_network_tf2_test.py Project: yeomju311/GRIN

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

Example #2

File: hourglass_network_tf2_test.py Project: yeomju311/GRIN

  def test_hourglass_104(self):

    net = hourglass.hourglass_104()
    self.assertEqual(hourglass.hourglass_depth(net), 104)

Example #3

File: hourglass_network_tf2_test.py Project: Asharib90/OCR

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

Example #4

File: hourglass_network_tf2_test.py Project: Asharib90/OCR

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

Example #5

 def test_hourglass_52(self):
     net = hourglass.hourglass_52(2)
     self.assertEqual(hourglass.hourglass_depth(net), 52)

Example #6

 def test_hourglass_20(self):
     net = hourglass.hourglass_20(2)
     self.assertEqual(hourglass.hourglass_depth(net), 20)