Skip to content

Commit 4e9d575

Browse files
committed
Add tensorflow->caffe mobilenet test in all-test.
1 parent 5a0cdda commit 4e9d575

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_conversion_imagenet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -811,8 +811,8 @@ def OnnxEmit(original_framework, architecture_name, architecture_path, weight_pa
811811
'inception_v3' : [CaffeEmit, CoreMLEmit, CntkEmit, KerasEmit, MXNetEmit, PytorchEmit, TensorflowEmit],
812812
'resnet_v1_152' : [CaffeEmit, CoreMLEmit, KerasEmit, MXNetEmit, PytorchEmit, TensorflowEmit], # TODO: CntkEmit
813813
'resnet_v2_152' : [CaffeEmit, CoreMLEmit, CntkEmit, KerasEmit, MXNetEmit, PytorchEmit, TensorflowEmit],
814-
'mobilenet_v1_1.0' : [CoreMLEmit, CntkEmit, KerasEmit, MXNetEmit, PytorchEmit, TensorflowEmit], # TODO: CaffeEmit(Crash)
815-
'mobilenet_v2_1.0_224' : [CoreMLEmit, CntkEmit, KerasEmit, MXNetEmit, PytorchEmit, TensorflowEmit], # TODO: CaffeEmit(Crash)
814+
'mobilenet_v1_1.0' : [CaffeEmit, CoreMLEmit, CntkEmit, KerasEmit, MXNetEmit, PytorchEmit, TensorflowEmit],
815+
'mobilenet_v2_1.0_224' : [CaffeEmit, CoreMLEmit, CntkEmit, KerasEmit, MXNetEmit, PytorchEmit, TensorflowEmit],
816816
'nasnet-a_large' : [MXNetEmit, PytorchEmit, TensorflowEmit], # TODO: KerasEmit(Slice Layer: https://blog.csdn.net/lujiandong1/article/details/54936185)
817817
'inception_resnet_v2' : [CaffeEmit, KerasEmit, MXNetEmit, PytorchEmit, TensorflowEmit], # CoremlEmit worked once, then always crashed
818818
},

0 commit comments

Comments
 (0)