Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleX
提交
e2fb8b2e
P
PaddleX
项目概览
PaddlePaddle
/
PaddleX
通知
138
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
43
列表
看板
标记
里程碑
合并请求
5
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleX
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
43
Issue
43
列表
看板
标记
里程碑
合并请求
5
合并请求
5
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
e2fb8b2e
编写于
6月 05, 2020
作者:
J
jiangjiajun
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix bug of export inference
上级
cf529fc1
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
2 addition
and
15 deletion
+2
-15
paddlex/cv/models/load_model.py
paddlex/cv/models/load_model.py
+1
-14
setup.py
setup.py
+1
-1
未找到文件。
paddlex/cv/models/load_model.py
浏览文件 @
e2fb8b2e
...
...
@@ -116,20 +116,7 @@ def fix_input_shape(info, fixed_input_shape=None):
resize
=
{
'ResizeByShort'
:
{}}
padding
=
{
'Padding'
:
{}}
if
info
[
'_Attributes'
][
'model_type'
]
==
'classifier'
:
crop_size
=
0
for
transform
in
info
[
'Transforms'
]:
if
'CenterCrop'
in
transform
:
crop_size
=
transform
[
'CenterCrop'
][
'crop_size'
]
break
assert
crop_size
==
fixed_input_shape
[
0
],
"fixed_input_shape must == CenterCrop:crop_size:{}"
.
format
(
crop_size
)
assert
crop_size
==
fixed_input_shape
[
1
],
"fixed_input_shape must == CenterCrop:crop_size:{}"
.
format
(
crop_size
)
if
crop_size
==
0
:
logging
.
warning
(
"fixed_input_shape must == input shape when trainning"
)
pass
else
:
resize
[
'ResizeByShort'
][
'short_size'
]
=
min
(
fixed_input_shape
)
resize
[
'ResizeByShort'
][
'max_size'
]
=
max
(
fixed_input_shape
)
...
...
setup.py
浏览文件 @
e2fb8b2e
...
...
@@ -19,7 +19,7 @@ long_description = "PaddleX. A end-to-end deeplearning model development toolkit
setuptools
.
setup
(
name
=
"paddlex"
,
version
=
'1.0.6'
,
version
=
'1.0.6
a4
'
,
author
=
"paddlex"
,
author_email
=
"paddlex@baidu.com"
,
description
=
long_description
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录