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