зеркало из https://github.com/microsoft/MMdnn.git
Merge pull request #76 from namizzz/master
caffe test table and download prototxt name
This commit is contained in:
Коммит
6aafb95d34
|
@ -36,7 +36,8 @@ class caffe_extractor(base_extractor):
|
|||
@classmethod
|
||||
def download(cls, architecture, path="./"):
|
||||
if cls.sanity_check(architecture):
|
||||
architecture_file = download_file(cls.architecture_map[architecture]['prototxt'], directory=path)
|
||||
prototxt_name = architecture + "-deploy.prototxt"
|
||||
architecture_file = download_file(cls.architecture_map[architecture]['prototxt'], directory=path ,local_fname=prototxt_name)
|
||||
if not architecture_file:
|
||||
return None
|
||||
|
||||
|
|
|
@ -64,7 +64,7 @@ class TestKit(object):
|
|||
'caffe' : {
|
||||
'alexnet' : lambda path : TestKit.ZeroCenter(path, 227, True),
|
||||
'vgg19' : lambda path : TestKit.ZeroCenter(path, 224, True),
|
||||
'inception_v1' : lambda path : TestKit.ZeroCenter(path, 227, True),
|
||||
'inception_v1' : lambda path : TestKit.ZeroCenter(path, 224, True),
|
||||
'resnet152' : lambda path : TestKit.ZeroCenter(path, 224, True),
|
||||
'squeezenet' : lambda path : TestKit.ZeroCenter(path, 227, False)
|
||||
},
|
||||
|
|
|
@ -313,11 +313,11 @@ class TestModels(CorrectnessTest):
|
|||
# 'imagenet1k-resnext-50' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
|
||||
},
|
||||
'caffe' : {
|
||||
'vgg19' : [KerasEmit],
|
||||
# 'alexnet' : [KerasEmit],
|
||||
'inception_v1' : [CntkEmit],
|
||||
'resnet152' : [CntkEmit],
|
||||
'squeezenet' : [CntkEmit]
|
||||
'vgg19' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
|
||||
'alexnet' : [CntkEmit],
|
||||
'inception_v1' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
|
||||
'resnet152' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
|
||||
'squeezenet' : [CntkEmit, PytorchEmit]
|
||||
},
|
||||
'tensorflow' : {
|
||||
'inception_v1' : [TensorflowEmit, KerasEmit, PytorchEmit], # TODO: CntkEmit
|
||||
|
|
Загрузка…
Ссылка в новой задаче