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_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_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_tiny_yolo_v2_opset1(self):
self._test_model(download_and_untar_once(
url=
"https://onnxzoo.blob.core.windows.net/models/opset_1/tiny_yolov2/tiny_yolov2.tar.gz",
member="*.onnx",
path="_models/onnx/tiny_yolov2_opset1.onnx"),
compare=False) # original ONNX is not runnable
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_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) # original ONNX graph is not runnable
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"),
compare=False
) # can't run the legacy ImageScaler op in original ONNX
def test_places205_alexnet(self):
model_path, prototxt_path = download_and_untar_once(
url="http://places.csail.mit.edu/model/placesCNN.tar.gz",
member=['*.caffemodel', '*deploy.prototxt'],
path=[
"_models/caffe/places_205_alexnet/places_205_alexnet.caffemodel",
"_models/caffe/places_205_alexnet/deploy.prototxt"
])
self._test_model(model_path=model_path, prototxt_path=prototxt_path)
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, source_shape=[1, 224, 224, 3])
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"))
def test_squeezenet(self):
path = download_and_untar_once(
url="http://download.tensorflow.org/models/tflite/model_zoo/upload_20180427/squeezenet_2018_04_27.tgz",
member="*.pb",
path="_models/tensorflow-pb/squeezenet_2018_04_27.pb")
self._test_network(path)
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)
