Пример #1
0
 def test_works_with_mobilenet_v2(self,):
     mobilenet = backbone.MobileNetBackbone('mobilenet_v2')
     model = mtp.MTP(mobilenet, 3, 6, 2, input_shape=(3, 100, 100))
     self._run(model)
Пример #2
0
 def test_works_with_resnet_152(self,):
     rn_152 = backbone.ResNetBackbone('resnet152')
     model = mtp.MTP(rn_152, 3, 6, 2, input_shape=(3, 100, 100))
     self._run(model)