Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
1e5be079
P
PaddleHub
项目概览
PaddlePaddle
/
PaddleHub
1 年多 前同步成功
通知
283
Star
12117
Fork
2091
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
200
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleHub
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
200
Issue
200
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
1e5be079
编写于
3月 11, 2020
作者:
W
wuzewu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fixed installation and loading issues when model have "-" in name
上级
666a91fa
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
14 addition
and
3 deletion
+14
-3
paddlehub/module/manager.py
paddlehub/module/manager.py
+14
-3
未找到文件。
paddlehub/module/manager.py
浏览文件 @
1e5be079
...
...
@@ -57,6 +57,8 @@ class LocalModuleManager(object):
desc
.
ParseFromString
(
fp
.
read
())
info
[
'version'
]
=
desc
.
attr
.
map
.
data
[
"module_info"
].
map
.
data
[
"version"
].
s
info
[
'name'
]
=
desc
.
attr
.
map
.
data
[
"module_info"
].
map
.
data
[
"name"
].
s
return
True
,
info
else
:
module_file
=
os
.
path
.
join
(
module_path
,
'module.py'
)
...
...
@@ -77,7 +79,7 @@ class LocalModuleManager(object):
version
=
_item
.
_version
break
sys
.
path
.
pop
(
0
)
return
True
,
{
'version'
:
version
}
return
True
,
{
'version'
:
version
,
'name'
:
_item
.
_name
}
logger
.
warning
(
"%s does not exist, the module will be reinstalled"
%
desc_pb_path
)
...
...
@@ -92,9 +94,13 @@ class LocalModuleManager(object):
for
sub_dir_name
in
os
.
listdir
(
self
.
local_modules_dir
):
sub_dir_path
=
os
.
path
.
join
(
self
.
local_modules_dir
,
sub_dir_name
)
if
os
.
path
.
isdir
(
sub_dir_path
):
if
"-"
in
sub_dir_path
:
new_sub_dir_path
=
sub_dir_path
.
replace
(
"-"
,
"_"
)
shutil
.
move
(
sub_dir_path
,
new_sub_dir_path
)
sub_dir_path
=
new_sub_dir_path
valid
,
info
=
self
.
check_module_valid
(
sub_dir_path
)
if
valid
:
module_name
=
sub_dir_name
module_name
=
info
[
'name'
]
self
.
modules_dict
[
module_name
]
=
(
sub_dir_path
,
info
[
'version'
])
return
self
.
modules_dict
...
...
@@ -175,6 +181,10 @@ class LocalModuleManager(object):
module_dir
=
os
.
path
.
join
(
_dir
,
file_names
[
0
])
for
index
,
file_name
in
enumerate
(
file_names
):
tar
.
extract
(
file_name
,
_dir
)
if
"-"
in
module_dir
:
new_module_dir
=
module_dir
.
replace
(
"-"
,
"_"
)
shutil
.
move
(
module_dir
,
new_module_dir
)
module_dir
=
new_module_dir
module_name
=
hub
.
Module
(
directory
=
module_dir
).
name
if
from_user_dir
:
...
...
@@ -198,7 +208,8 @@ class LocalModuleManager(object):
"w"
)
as
fp
:
fp
.
write
(
md5_value
)
save_path
=
os
.
path
.
join
(
MODULE_HOME
,
module_name
)
save_path
=
os
.
path
.
join
(
MODULE_HOME
,
module_name
.
replace
(
"-"
,
"_"
))
if
save_path
!=
module_dir
:
if
os
.
path
.
exists
(
save_path
):
shutil
.
rmtree
(
save_path
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录