def test_mobilenet_v2_1_0_224(self):
path = download_and_untar_once(
url=
"http://download.tensorflow.org/models/tflite_11_05_08/mobilenet_v2_1.0_224.tgz",
member="*.pb",
path="_models/tensorflow-pb/mobilenet_v2_1.0_224.pb")
self._test_network(path, 224)
def test_inception_resnet_v2(self):
path = download_and_untar_once(
url=
"http://download.tensorflow.org/models/tflite/model_zoo/upload_20180427/inception_resnet_v2_2018_04_27.tgz",
member="*.pb",
path="_models/tensorflow-pb/inception_resnet_v2_2018_04_27.pb")
self._test_network(path, 299)
def test_mobilenet_v1_1_0_128(self):
path = download_and_untar_once(
url=
"http://download.tensorflow.org/models/mobilenet_v1_2018_02_22/mobilenet_v1_1.0_128.tgz",
member="*.pb",
path="_models/tensorflow-pb/mobilenet_v1_1.0_128.pb")
self._test_network(path, 128)
def test_emotion_ferplus_opset1(self):
self._test_model(download_and_untar_once(
url=
"https://onnxzoo.blob.core.windows.net/models/opset_2/emotion_ferplus/emotion_ferplus.tar.gz",
member="*.onnx",
path="_models/onnx/emotion_ferplus_opset1.onnx"),
compare=False)
def test_nasnet_large(self):
path = download_and_untar_once(
url=
"http://download.tensorflow.org/models/tflite/model_zoo/upload_20180427/nasnet_large_2018_04_27.tgz",
member="*.pb",
path="_models/tensorflow-pb/nasnet_large_2018_04_27.pb")
self._test_network(path, 331)
def test_caffenet(self):
self._test_model(
download_and_untar_once(
url=
"https://s3.amazonaws.com/download.onnx/models/opset_8/bvlc_reference_caffenet.tar.gz",
member="*.onnx",
path="_models/onnx/bvlc_reference_caffenet.onnx"))
def test_zfnet512(self):
self._test_model(
download_and_untar_once(
url=
"https://s3.amazonaws.com/download.onnx/models/opset_8/zfnet512.tar.gz",
member="*.onnx",
path="_models/onnx/zfnet512.onnx"))
def test_mnist(self):
self._test_model(download_and_untar_once(
url=
"https://onnxzoo.blob.core.windows.net/models/opset_8/mnist/mnist.tar.gz",
member="*.onnx",
path="_models/onnx/mnist.onnx"),
compare=False)
def test_inception(self):
self._test_model(
download_and_untar_once(
url=
"https://s3.amazonaws.com/download.onnx/models/opset_8/inception_v1.tar.gz",
member="*.onnx",
path="_models/onnx/inception_v1.onnx"))
def test_tiny_yolo_v2_opset7(self):
self._test_model(download_and_untar_once(
url=
"https://onnxzoo.blob.core.windows.net/models/opset_7/tiny_yolov2/tiny_yolov2.tar.gz",
member="*.onnx",
path="_models/onnx/tiny_yolov2_opset7.onnx"),
source_shape="{'image': (FLOAT, [2, 3, 416, 416])}",
compare=False)
def test_inception_v4(self):
self._test_model(download_and_untar_once(
url="http://download.tensorflow.org/models/tflite/model_zoo/upload_20180427/inception_v4_2018_04_27.tgz",
member="*.tflite",
path="_models/tensorflow-lite/inception_v4_2018_04_27.tflite"))
def test_resnet_v2_101_299(self):
self._test_model(download_and_untar_once(
url="http://download.tensorflow.org/models/tflite_11_05_08/resnet_v2_101.tgz",
member="*.tflite",
path="_models/tensorflow-lite/resnet_v2_101.tflite"))
def test_mobilenet_v1_0_25_128_quant(self):
self._test_model(download_and_untar_once(
url="http://download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_0.25_128_quant.tgz",
member="*.tflite",
path="_models/tensorflow-lite/mobilenet_v1_0.25_128_quant.tflite"))
def test_inception_v1_224_quant(self):
self._test_model(download_and_untar_once(
url="http://download.tensorflow.org/models/inception_v1_224_quant_20181026.tgz",
member="*.tflite",
path="_models/tensorflow-lite/inception_v1_224_quant_20181026.tflite"))
def test_mobilenet_v2_1_0_224(self):
self._test_model(download_and_untar_once(
url="http://download.tensorflow.org/models/tflite_11_05_08/mobilenet_v2_1.0_224.tgz",
member="*.tflite",
path="_models/tensorflow-lite/mobilenet_v2_1.0_224.tflite"))
def test_mnasnet_0_5_224(self):
self._test_model(download_and_untar_once(
url="http://download.tensorflow.org/models/tflite/mnasnet_0.5_224_09_07_2018.tgz",
member="*.tflite",
path="_models/tensorflow-lite/mnasnet_0.5_224_09_07_2018.tflite"))
