microsoft
/
MMdnn
зеркало из https://github.com/microsoft/MMdnn.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

Finish MXNet -> other pytest.

This commit is contained in:
Kit 2018-02-07 18:03:58 +08:00
Родитель 059988a4a8
Коммит 3a81b7adce
3 изменённых файлов: 26 добавлений и 21 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

19
mmdnn/conversion/examples/imagenet_test.py
Просмотреть файл

@ -91,15 +91,16 @@ class TestKit(object):
},
'mxnet' : {
'vgg16' : lambda path : TestKit.ZeroCenter(path, 224, False),
'vgg19' : lambda path : TestKit.ZeroCenter(path, 224, False),
'resnet' : lambda path : TestKit.Identity(path, 224, True),
'squeezenet_v1.0' : lambda path : TestKit.ZeroCenter(path, 224, False),
'squeezenet_v1.1' : lambda path : TestKit.ZeroCenter(path, 224, False),
'inception_bn' : lambda path : TestKit.Identity(path, 224, False),
'resnet152-11k' : lambda path : TestKit.Identity(path, 224, True),
'resnext' : lambda path : TestKit.Identity(path, 224, False),
'imagenet1k-resnext-50' : lambda path : TestKit.Identity(path, 224, False)
'vgg16' : lambda path : TestKit.ZeroCenter(path, 224, False),
'vgg19' : lambda path : TestKit.ZeroCenter(path, 224, False),
'resnet' : lambda path : TestKit.Identity(path, 224, True),
'squeezenet_v1.0' : lambda path : TestKit.ZeroCenter(path, 224, False),
'squeezenet_v1.1' : lambda path : TestKit.ZeroCenter(path, 224, False),
'imagenet1k-inception-bn' : lambda path : TestKit.Identity(path, 224, False),
'imagenet1k-resnet-152' : lambda path : TestKit.Identity(path, 224, True),
'resnext' : lambda path : TestKit.Identity(path, 224, False),
'imagenet1k-resnext-50' : lambda path : TestKit.Identity(path, 224, False),
'imagenet1k-resnext-101-64x4d' : lambda path : TestKit.Identity(path, 224, False)
},
'pytorch' : {

21
mmdnn/conversion/examples/mxnet/extractor.py
Просмотреть файл

@ -4,7 +4,6 @@
#----------------------------------------------------------------------------------------------
from __future__ import absolute_import
import os
from mmdnn.conversion.examples.imagenet_test import TestKit
from mmdnn.conversion.examples.extractor import base_extractor
from mmdnn.conversion.common.utils import download_file
@ -38,20 +37,20 @@ class mxnet_extractor(base_extractor):
'params' : _base_model_url+'imagenet/resnext/101-layers/resnext-101-0000.params'},
'imagenet1k-resnext-101-64x4d' : {'symbol' : _base_model_url+'imagenet/resnext/101-layers/resnext-101-64x4d-symbol.json',
'params' : _base_model_url+'imagenet/resnext/101-layers/resnext-101-64x4d-0000.params'},
'imagenet11k-resnet-152' : {'symbol' : _base_model_url+'imagenet-11k/resnet-152/resnet-152-symbol.json',
'params' : _base_model_url+'imagenet-11k/resnet-152/resnet-152-0000.params'},
'imagenet11k-place365ch-resnet-152' : {'symbol' : _base_model_url+'imagenet-11k-place365-ch/resnet-152-symbol.json',
'params' : _base_model_url+'imagenet-11k-place365-ch/resnet-152-0000.params'},
'imagenet11k-place365ch-resnet-50' : {'symbol' : _base_model_url+'imagenet-11k-place365-ch/resnet-50-symbol.json',
'params' : _base_model_url+'imagenet-11k-place365-ch/resnet-50-0000.params'},
'imagenet11k-resnet-152' : {'symbol' : _base_model_url+'imagenet-11k/resnet-152/resnet-152-symbol.json',
'params' : _base_model_url+'imagenet-11k/resnet-152/resnet-152-0000.params'},
'imagenet11k-place365ch-resnet-152' : {'symbol' : _base_model_url+'imagenet-11k-place365-ch/resnet-152-symbol.json',
'params' : _base_model_url+'imagenet-11k-place365-ch/resnet-152-0000.params'},
'imagenet11k-place365ch-resnet-50' : {'symbol' : _base_model_url+'imagenet-11k-place365-ch/resnet-50-symbol.json',
'params' : _base_model_url+'imagenet-11k-place365-ch/resnet-50-0000.params'},
'vgg19' : {'symbol' : _base_model_url+'imagenet/vgg/vgg19-symbol.json',
'params' : _base_model_url+'imagenet/vgg/vgg19-0000.params'},
'vgg16' : {'symbol' : _base_model_url+'imagenet/vgg/vgg16-symbol.json',
'params' : _base_model_url+'imagenet/vgg/vgg16-0000.params'},
'squeezenet_v1.0' : {'symbol' : _base_model_url+'imagenet/squeezenet/squeezenet_v1.0-symbol.json',
'params' : _base_model_url+'imagenet/squeezenet/squeezenet_v1.0-0000.params'},
'squeezenet_v1.1' : {'symbol' : _base_model_url+'imagenet/squeezenet/squeezenet_v1.1-symbol.json',
'params' : _base_model_url+'imagenet/squeezenet/squeezenet_v1.1-0000.params'}
'squeezenet_v1.0' : {'symbol' : _base_model_url+'imagenet/squeezenet/squeezenet_v1.0-symbol.json',
'params' : _base_model_url+'imagenet/squeezenet/squeezenet_v1.0-0000.params'},
'squeezenet_v1.1' : {'symbol' : _base_model_url+'imagenet/squeezenet/squeezenet_v1.1-symbol.json',
'params' : _base_model_url+'imagenet/squeezenet/squeezenet_v1.1-0000.params'}
}

7
tests/test_conversion_imagenet.py
Просмотреть файл

@ -235,7 +235,12 @@ class TestModels(CorrectnessTest):
'nasnet' : [TensorflowEmit, KerasEmit],
},
'mxnet' : {
'vgg19' : [CntkEmit, TensorflowEmit, KerasEmit],
'vgg19' : [CntkEmit, TensorflowEmit, KerasEmit],
'imagenet1k-inception-bn' : [CntkEmit, TensorflowEmit, KerasEmit],
'imagenet1k-resnet-152' : [CntkEmit, TensorflowEmit, KerasEmit],
'squeezenet_v1.1' : [CntkEmit, TensorflowEmit, KerasEmit],
'imagenet1k-resnext-101-64x4d' : [TensorflowEmit], # TODO: CntkEmit
'imagenet1k-resnext-50' : [TensorflowEmit, KerasEmit], # TODO: CntkEmit
}
}