1from __future__ import absolute_import 2from __future__ import print_function 3 4import os 5import sys 6from conversion_imagenet import TestModels 7 8def get_test_table(): 9 return { 'tensorflow' : 10 { 11 'vgg19' : [ 12 TestModels.onnx_emit, 13 TestModels.caffe_emit, 14 TestModels.cntk_emit, 15 TestModels.coreml_emit, 16 TestModels.keras_emit, 17 TestModels.mxnet_emit, 18 TestModels.pytorch_emit, 19 TestModels.tensorflow_emit 20 ] 21 } 22 } 23 24def test_tensorflow(): 25 test_table = get_test_table() 26 tester = TestModels(test_table) 27 tester._test_function('tensorflow', tester.tensorflow_parse) 28 29 30if __name__ == "__main__": 31 test_tensorflow() 32