Ejemplo n.º 1
0
 def test_AgeXceptionTiny(self):
   net = AgeXceptionTiny()
   test_net(net, create_fake_input_tensor(112), [1])
Ejemplo n.º 2
0
 def test_MobilenetV2(self):
   net = MobilenetV2()
   test_net(net, create_fake_input_tensor(112), [1])
Ejemplo n.º 3
0
 def test_Densenet_4_4_DEX(self):
   net = Densenet_4_4_DEX()
   test_net(net, create_fake_input_tensor(112), [1])
Ejemplo n.º 4
0
 def test_XceptionTiny(self):
   net = XceptionTiny()
   test_net(net, create_fake_input_tensor(112), [1, 7, 7, 512])
Ejemplo n.º 5
0
 def test_Densenet_4_5_FeatureExtractor(self):
   net = Densenet_4_5_FeatureExtractor()
   test_net(net, create_fake_input_tensor(112), [1, 7, 7, 512])
Ejemplo n.º 6
0
 def test_AgeGenderXceptionTiny(self):
     net = AgeGenderXceptionTiny()
     test_net(net, create_fake_input_tensor(112), ([1], [1, 2]))