Merge pull request #76 from namizzz/master

caffe test table and download prototxt name
This commit is contained in:
Kit CHEN 2018-02-09 13:56:28 +08:00 коммит произвёл GitHub
Родитель c4aba456d0 58c9e22bf0
Коммит 6aafb95d34
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
3 изменённых файлов: 8 добавлений и 7 удалений

Просмотреть файл

@ -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