Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
569a9d39
P
PaddleHub
项目概览
PaddlePaddle
/
PaddleHub
大约 1 年 前同步成功
通知
280
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
569a9d39
编写于
4月 30, 2020
作者:
W
wuzewu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update image classification module
上级
e6de8757
变更
10
显示空白变更内容
内联
并排
Showing
10 changed file
with
16 addition
and
16 deletion
+16
-16
hub_module/modules/image/classification/efficientnetb0_small_imagenet/module.py
...ge/classification/efficientnetb0_small_imagenet/module.py
+2
-2
hub_module/modules/image/classification/efficientnetb0_small_imagenet/processor.py
...classification/efficientnetb0_small_imagenet/processor.py
+1
-1
hub_module/modules/image/classification/fix_resnext101_32x48d_wsl_imagenet/module.py
...assification/fix_resnext101_32x48d_wsl_imagenet/module.py
+3
-3
hub_module/modules/image/classification/fix_resnext101_32x48d_wsl_imagenet/processor.py
...ification/fix_resnext101_32x48d_wsl_imagenet/processor.py
+1
-1
hub_module/modules/image/classification/res2net101_vd_26w_4s_imagenet/module.py
...ge/classification/res2net101_vd_26w_4s_imagenet/module.py
+2
-2
hub_module/modules/image/classification/res2net101_vd_26w_4s_imagenet/processor.py
...classification/res2net101_vd_26w_4s_imagenet/processor.py
+1
-1
hub_module/modules/image/classification/resnet18_vd_imagenet/module.py
...dules/image/classification/resnet18_vd_imagenet/module.py
+2
-2
hub_module/modules/image/classification/resnet18_vd_imagenet/processor.py
...es/image/classification/resnet18_vd_imagenet/processor.py
+1
-1
hub_module/modules/image/classification/se_resnet18_vd_imagenet/module.py
...es/image/classification/se_resnet18_vd_imagenet/module.py
+2
-2
hub_module/modules/image/classification/se_resnet18_vd_imagenet/processor.py
...image/classification/se_resnet18_vd_imagenet/processor.py
+1
-1
未找到文件。
hub_module/modules/image/classification/efficientnetb0_small_imagenet/module.py
浏览文件 @
569a9d39
...
...
@@ -21,8 +21,8 @@ from efficientnetb0_small_imagenet.efficientnet import EfficientNetB0_small
@
moduleinfo
(
name
=
"efficientnetb0_small_imagenet"
,
type
=
"CV/image_classification"
,
author
=
"
baidu-vis
"
,
author_email
=
""
,
author
=
"
paddlepaddle
"
,
author_email
=
"
paddle-dev@baidu.com
"
,
summary
=
"ResNet18vd is a image classfication model, this module is trained with imagenet datasets."
,
version
=
"1.0.0"
)
...
...
hub_module/modules/image/classification/efficientnetb0_small_imagenet/processor.py
浏览文件 @
569a9d39
...
...
@@ -49,7 +49,7 @@ def postprocess(data_out, label_list, top_k):
output_i
=
{}
indexs
=
np
.
argsort
(
result_i
)[::
-
1
][
0
:
top_k
]
for
index
in
indexs
:
label
=
label_list
[
index
]
label
=
label_list
[
index
]
.
split
(
','
)[
0
]
output_i
[
label
]
=
float
(
result_i
[
index
])
output
.
append
(
output_i
)
return
output
hub_module/modules/image/classification/fix_resnext101_32x48d_wsl_imagenet/module.py
浏览文件 @
569a9d39
...
...
@@ -21,15 +21,15 @@ from fix_resnext101_32x48d_wsl_imagenet.resnext101_wsl import Fix_ResNeXt101_32x
@
moduleinfo
(
name
=
"fix_resnext101_32x48d_wsl_imagenet"
,
type
=
"CV/image_classification"
,
author
=
"
baidu-vis
"
,
author_email
=
""
,
author
=
"
paddlepaddle
"
,
author_email
=
"
paddle-dev@baidu.com
"
,
summary
=
"fix_resnext101_32x48d_wsl is a image classfication model, this module is trained with imagenet datasets."
,
version
=
"1.0.0"
)
class
FixResnext10132x48dwslImagenet
(
hub
.
Module
):
def
_initialize
(
self
):
self
.
default_pretrained_model_path
=
os
.
path
.
join
(
self
.
directory
,
"
fix_resnext101_32x48d_wsl_imagenet_
model"
)
self
.
directory
,
"model"
)
label_file
=
os
.
path
.
join
(
self
.
directory
,
"label_list.txt"
)
with
open
(
label_file
,
'r'
,
encoding
=
'utf-8'
)
as
file
:
self
.
label_list
=
file
.
read
().
split
(
"
\n
"
)[:
-
1
]
...
...
hub_module/modules/image/classification/fix_resnext101_32x48d_wsl_imagenet/processor.py
浏览文件 @
569a9d39
...
...
@@ -49,7 +49,7 @@ def postprocess(data_out, label_list, top_k):
output_i
=
{}
indexs
=
np
.
argsort
(
result_i
)[::
-
1
][
0
:
top_k
]
for
index
in
indexs
:
label
=
label_list
[
index
]
label
=
label_list
[
index
]
.
split
(
','
)[
0
]
output_i
[
label
]
=
float
(
result_i
[
index
])
output
.
append
(
output_i
)
return
output
hub_module/modules/image/classification/res2net101_vd_26w_4s_imagenet/module.py
浏览文件 @
569a9d39
...
...
@@ -21,8 +21,8 @@ from res2net101_vd_26w_4s_imagenet.res2net_vd import Res2Net101_vd_26w_4s
@
moduleinfo
(
name
=
"res2net101_vd_26w_4s_imagenet"
,
type
=
"CV/image_classification"
,
author
=
"
baidu-vis
"
,
author_email
=
""
,
author
=
"
paddlepaddle
"
,
author_email
=
"
paddle-dev@baidu.com
"
,
summary
=
"res2net101_vd_26w_4s is a image classfication model, this module is trained with imagenet datasets."
,
version
=
"1.0.0"
)
...
...
hub_module/modules/image/classification/res2net101_vd_26w_4s_imagenet/processor.py
浏览文件 @
569a9d39
...
...
@@ -49,7 +49,7 @@ def postprocess(data_out, label_list, top_k):
output_i
=
{}
indexs
=
np
.
argsort
(
result_i
)[::
-
1
][
0
:
top_k
]
for
index
in
indexs
:
label
=
label_list
[
index
]
label
=
label_list
[
index
]
.
split
(
','
)[
0
]
output_i
[
label
]
=
float
(
result_i
[
index
])
output
.
append
(
output_i
)
return
output
hub_module/modules/image/classification/resnet18_vd_imagenet/module.py
浏览文件 @
569a9d39
...
...
@@ -21,8 +21,8 @@ from resnet18_vd_imagenet.resnet_vd import ResNet18_vd
@
moduleinfo
(
name
=
"resnet18_vd_imagenet"
,
type
=
"CV/image_classification"
,
author
=
"
baidu-vis
"
,
author_email
=
""
,
author
=
"
paddlepaddle
"
,
author_email
=
"
paddle-dev@baidu.com
"
,
summary
=
"ResNet18vd is a image classfication model, this module is trained with imagenet datasets."
,
version
=
"1.0.0"
)
...
...
hub_module/modules/image/classification/resnet18_vd_imagenet/processor.py
浏览文件 @
569a9d39
...
...
@@ -49,7 +49,7 @@ def postprocess(data_out, label_list, top_k):
output_i
=
{}
indexs
=
np
.
argsort
(
result_i
)[::
-
1
][
0
:
top_k
]
for
index
in
indexs
:
label
=
label_list
[
index
]
label
=
label_list
[
index
]
.
split
(
','
)[
0
]
output_i
[
label
]
=
float
(
result_i
[
index
])
output
.
append
(
output_i
)
return
output
hub_module/modules/image/classification/se_resnet18_vd_imagenet/module.py
浏览文件 @
569a9d39
...
...
@@ -21,8 +21,8 @@ from se_resnet18_vd_imagenet.se_resnet import SE_ResNet18_vd
@
moduleinfo
(
name
=
"se_resnet18_vd_imagenet"
,
type
=
"CV/image_classification"
,
author
=
"
baidu-vis
"
,
author_email
=
""
,
author
=
"
paddlepaddle
"
,
author_email
=
"
paddle-dev@baidu.com
"
,
summary
=
"SE_ResNet18_vd is a image classfication model, this module is trained with imagenet datasets."
,
version
=
"1.0.0"
)
...
...
hub_module/modules/image/classification/se_resnet18_vd_imagenet/processor.py
浏览文件 @
569a9d39
...
...
@@ -49,7 +49,7 @@ def postprocess(data_out, label_list, top_k):
output_i
=
{}
indexs
=
np
.
argsort
(
result_i
)[::
-
1
][
0
:
top_k
]
for
index
in
indexs
:
label
=
label_list
[
index
]
label
=
label_list
[
index
]
.
split
(
','
)[
0
]
output_i
[
label
]
=
float
(
result_i
[
index
])
output
.
append
(
output_i
)
return
output
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录