def test_resnet18v1(self):
# ONNX has some problem with running the BatchNorm version 7 and it cannot convert it to newer version
self._test_model(download_once(
url=
"https://s3.amazonaws.com/onnx-model-zoo/resnet/resnet18v1/resnet18v1.onnx",
path="_models/onnx/"),
run=False)
def test_squeezenet1_1(self):
self._test_model(
download_once(
url=
"https://s3.amazonaws.com/onnx-model-zoo/squeezenet/squeezenet1.1/squeezenet1.1.onnx",
path="./_models/onnx/"))
def test_vgg16(self):
self._test_model(
download_once(
url=
"https://s3.amazonaws.com/onnx-model-zoo/vgg/vgg16/vgg16.onnx",
path="_models/onnx/"))
def test_resnet101v2(self):
self._test_model(download_once(
url=
"https://s3.amazonaws.com/onnx-model-zoo/resnet/resnet101v2/resnet101v2.onnx",
path="_models/onnx/"),
run=False)
def test_mobilenetv2_1_0(self):
self._test_model(
download_once(
url=
"https://s3.amazonaws.com/onnx-model-zoo/mobilenet/mobilenetv2-1.0/mobilenetv2-1.0.onnx",
path="_models/onnx/"))
def test_arcface(self):
self._test_model(download_once(
url=
"https://s3.amazonaws.com/onnx-model-zoo/arcface/resnet100/resnet100.onnx",
path="_models/onnx/arcface_resnet100.onnx"),
compare=False)