Exemple #1
0
 def test_save_load_forward(self):
   net = FPN3Stagev3_64_128_256()
   forward = lambda: net.forward(create_fake_input_tensor(224), 1, 224)
   test_net_save_load_forward(net, forward)
Exemple #2
0
 def test_MobilenetV2(self):
   net = MobilenetV2()
   test_net(net, create_fake_input_tensor(112), [1])
Exemple #3
0
 def test_AgeXceptionTiny(self):
   net = AgeXceptionTiny()
   test_net(net, create_fake_input_tensor(112), [1])
Exemple #4
0
 def test_Densenet_4_4_DEX(self):
   net = Densenet_4_4_DEX()
   test_net(net, create_fake_input_tensor(112), [1])
Exemple #5
0
 def test_save_load_forward(self):
     net = FPN2Stage_256_512()
     forward = lambda: net.forward(create_fake_input_tensor(224), 1, 224)
     test_net_save_load_forward(net, forward)
Exemple #6
0
 def test_XceptionTiny(self):
   net = XceptionTiny()
   test_net(net, create_fake_input_tensor(112), [1, 7, 7, 512])
Exemple #7
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])
Exemple #8
0
 def test_AgeGenderXceptionTiny(self):
     net = AgeGenderXceptionTiny()
     test_net(net, create_fake_input_tensor(112), ([1], [1, 2]))