提交 c8c287fc 编写于 作者: S sunyanfang01

update command.py

上级 737b5c01
...@@ -49,4 +49,4 @@ load_model = cv.models.load_model ...@@ -49,4 +49,4 @@ load_model = cv.models.load_model
datasets = cv.datasets datasets = cv.datasets
log_level = 2 log_level = 2
__version__ = '0.1.9.github' __version__ = '0.2.0.github'
...@@ -71,6 +71,8 @@ def main(): ...@@ -71,6 +71,8 @@ def main():
assert len( assert len(
fixed_input_shape fixed_input_shape
) == 2, "len of fixed input shape must == 2, such as [224,224]" ) == 2, "len of fixed input shape must == 2, such as [224,224]"
else:
fixed_input_shape = None
model = pdx.load_model(args.model_dir, fixed_input_shape) model = pdx.load_model(args.model_dir, fixed_input_shape)
model.export_inference_model(args.save_dir) model.export_inference_model(args.save_dir)
......
...@@ -19,7 +19,7 @@ long_description = "PaddleX. A end-to-end deeplearning model development toolkit ...@@ -19,7 +19,7 @@ long_description = "PaddleX. A end-to-end deeplearning model development toolkit
setuptools.setup( setuptools.setup(
name="paddlex", name="paddlex",
version='0.1.9', version='0.2.0',
author="paddlex", author="paddlex",
author_email="paddlex@baidu.com", author_email="paddlex@baidu.com",
description=long_description, description=long_description,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册