IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

MO cannot convert Caffe model from PyPI "openvino-dev" distribution (#5239)

Browse Source 
* MO cannot convert Caffe model from PyPI "openvino-dev" distribution

* use os.pardir instead of '..'

* remove mo subdir
This commit is contained in:
Sergey Lyubimtsev
2021-04-16 13:35:25 +03:00
committed by GitHub
parent e170785c1f
commit 8013d2f7ba
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
model-optimizer/mo/utils/cli_parser.py
View File

@@ -428,12 +428,12 @@ def get_caffe_cli_parser(parser: argparse.ArgumentParser = None):
caffe_group.add_argument('--caffe_parser_path',
help='Path to Python Caffe* parser generated from caffe.proto',
type=str,
default=os.path.join(os.path.dirname(sys.argv[0]), 'mo', 'front', 'caffe', 'proto'),
default=os.path.join(os.path.dirname(__file__), os.pardir, 'front', 'caffe', 'proto'),
action=CanonicalizePathCheckExistenceAction)
caffe_group.add_argument('-k',
help='Path to CustomLayersMapping.xml to register custom layers',
type=str,
default=os.path.join(os.path.dirname(sys.argv[0]), 'extensions', 'front', 'caffe',
default=os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, 'extensions', 'front', 'caffe',
'CustomLayersMapping.xml'),
action=CanonicalizePathCheckExistenceAction)
caffe_group.add_argument('--mean_file', '-mf',