Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
23575238
P
PaddleHub
项目概览
PaddlePaddle
/
PaddleHub
大约 1 年 前同步成功
通知
282
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看板
提交
23575238
编写于
4月 29, 2020
作者:
W
wuzewu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix docs
上级
c6553c45
变更
21
隐藏空白更改
内联
并排
Showing
21 changed file
with
63 addition
and
97 deletion
+63
-97
hub_module/modules/image/object_detection/faster_rcnn_resnet50_coco2017/README.md
.../object_detection/faster_rcnn_resnet50_coco2017/README.md
+3
-7
hub_module/modules/image/object_detection/faster_rcnn_resnet50_fpn_coco2017/README.md
...ect_detection/faster_rcnn_resnet50_fpn_coco2017/README.md
+3
-7
hub_module/modules/image/object_detection/retinanet_resnet50_fpn_coco2017/README.md
...bject_detection/retinanet_resnet50_fpn_coco2017/README.md
+3
-7
hub_module/modules/image/object_detection/ssd_mobilenet_v1_pascal/README.md
.../image/object_detection/ssd_mobilenet_v1_pascal/README.md
+3
-7
hub_module/modules/image/object_detection/ssd_vgg16_300_coco2017/README.md
...s/image/object_detection/ssd_vgg16_300_coco2017/README.md
+3
-7
hub_module/modules/image/object_detection/ssd_vgg16_512_coco2017/README.md
...s/image/object_detection/ssd_vgg16_512_coco2017/README.md
+3
-7
hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/README.md
...mage/object_detection/yolov3_darknet53_coco2017/README.md
+4
-8
hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/module.py
...mage/object_detection/yolov3_darknet53_coco2017/module.py
+4
-2
hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/yolo_head.py
...e/object_detection/yolov3_darknet53_coco2017/yolo_head.py
+1
-1
hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/README.md
...ge/object_detection/yolov3_darknet53_pedestrian/README.md
+4
-8
hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/module.py
...ge/object_detection/yolov3_darknet53_pedestrian/module.py
+4
-2
hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/yolo_head.py
...object_detection/yolov3_darknet53_pedestrian/yolo_head.py
+1
-1
hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/README.md
...mage/object_detection/yolov3_darknet53_vehicles/README.md
+4
-8
hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/module.py
...mage/object_detection/yolov3_darknet53_vehicles/module.py
+4
-2
hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/yolo_head.py
...e/object_detection/yolov3_darknet53_vehicles/yolo_head.py
+1
-1
hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/README.md
...e/object_detection/yolov3_mobilenet_v1_coco2017/README.md
+4
-8
hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/module.py
...e/object_detection/yolov3_mobilenet_v1_coco2017/module.py
+4
-2
hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/yolo_head.py
...bject_detection/yolov3_mobilenet_v1_coco2017/yolo_head.py
+1
-1
hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/README.md
...image/object_detection/yolov3_resnet34_coco2017/README.md
+4
-8
hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py
...image/object_detection/yolov3_resnet34_coco2017/module.py
+4
-2
hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/yolo_head.py
...ge/object_detection/yolov3_resnet34_coco2017/yolo_head.py
+1
-1
未找到文件。
hub_module/modules/image/object_detection/faster_rcnn_resnet50_coco2017/README.md
浏览文件 @
23575238
```
shell
$
hub
install
faster_rcnn_resnet50_coco2017
==
1.1.0
```
## 命令行预测
```
hub run faster_rcnn_resnet50_coco2017 --input_path "/PATH/TO/IMAGE"
$
hub run faster_rcnn_resnet50_coco2017 --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -22,7 +18,7 @@ def context(num_classes=81,
**参数**
*
num
\_
classes (int): 类别数;
*
trainable(bool):
将参数的trainable 属性设为trainable
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
phase (str): 可选值为 'train'/'predict','trian' 用于训练,'predict' 用于预测。
...
...
@@ -35,7 +31,7 @@ def context(num_classes=81,
*
im
\_
info (Variable): 图像缩放信息
*
gt
\_
class (Variable): 检测框类别
*
gt
\_
box (Variable): 检测框坐标
*
is
\_
crowd (Variable):
*
is
\_
crowd (Variable):
单个框内是否包含多个物体
当 phase 为 'predict'时,包含:
*
image (Variable): 图像变量
*
im
\_
size (Variable): 图像的尺寸
...
...
hub_module/modules/image/object_detection/faster_rcnn_resnet50_fpn_coco2017/README.md
浏览文件 @
23575238
```
shell
$
hub
install
faster_rcnn_resnet50_fpn_coco2017
==
1.0.0
```
## 命令行预测
```
hub run faster_rcnn_resnet50_fpn_coco2017 --input_path "/PATH/TO/IMAGE"
$
hub run faster_rcnn_resnet50_fpn_coco2017 --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -22,7 +18,7 @@ def context(num_classes=81,
**参数**
*
num
\_
classes (int): 类别数;
*
trainable(bool):
将参数的trainable 属性设为trainable
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
phase (str): 可选值为 'train'/'predict','trian' 用于训练,'predict' 用于预测。
...
...
@@ -35,7 +31,7 @@ def context(num_classes=81,
*
im
\_
info (Variable): 图像缩放信息
*
gt
\_
class (Variable): 检测框类别
*
gt
\_
box (Variable): 检测框坐标
*
is
\_
crowd (Variable):
*
is
\_
crowd (Variable):
单个框内是否包含多个物体
当 phase 为 'predict'时,包含:
*
image (Variable): 图像变量
*
im
\_
size (Variable): 图像的尺寸
...
...
hub_module/modules/image/object_detection/retinanet_resnet50_fpn_coco2017/README.md
浏览文件 @
23575238
```
shell
$
hub
install
retinanet_resnet50_fpn_coco2017
==
1.0.0
```
## 命令行预测
```
hub run retinanet_resnet50_fpn_coco2017 --input_path "/PATH/TO/IMAGE"
$
hub run retinanet_resnet50_fpn_coco2017 --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -16,11 +12,11 @@ def context(trainable=True,
get_prediction=False)
```
特征提取
,用于迁移学习。
提取特征
,用于迁移学习。
**参数**
*
trainable(bool):
将参数的trainable属性设为trainable
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
get
\_
prediction (bool): 是否执行预测。
...
...
hub_module/modules/image/object_detection/ssd_mobilenet_v1_pascal/README.md
浏览文件 @
23575238
```
shell
$
hub
install
ssd_mobilenet_v1_pascal
==
1.1.0
```
## 命令行预测
```
hub run ssd_mobilenet_v1_pascal --input_path "/PATH/TO/IMAGE"
$
hub run ssd_mobilenet_v1_pascal --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -16,11 +12,11 @@ def context(trainable=True,
get_prediction=False)
```
特征提取
,用于迁移学习。
提取特征
,用于迁移学习。
**参数**
*
trainable(bool):
设置参数的 trainable 属性
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
get
\_
prediction (bool): 是否执行预测。
...
...
hub_module/modules/image/object_detection/ssd_vgg16_300_coco2017/README.md
浏览文件 @
23575238
```
shell
$
hub
install
ssd_vgg16_300_coco2017
==
1.0.0
```
## 命令行预测
```
hub run ssd_vgg16_300_coco2017 --input_path "/PATH/TO/IMAGE"
$
hub run ssd_vgg16_300_coco2017 --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -16,11 +12,11 @@ def context(trainable=True,
get_prediction=False)
```
特征提取
,用于迁移学习。
提取特征
,用于迁移学习。
**参数**
*
trainable(bool):
设置参数的 trainable 属性
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
get
\_
prediction (bool): 是否执行预测。
...
...
hub_module/modules/image/object_detection/ssd_vgg16_512_coco2017/README.md
浏览文件 @
23575238
```
shell
$
hub
install
ssd_vgg16_512_coco2017
==
1.0.0
```
## 命令行预测
```
hub run ssd_vgg16_512_coco2017 --input_path "/PATH/TO/IMAGE"
$
hub run ssd_vgg16_512_coco2017 --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -16,11 +12,11 @@ def context(trainable=True,
get_prediction=False)
```
特征提取
,用于迁移学习。
提取特征
,用于迁移学习。
**参数**
*
trainable(bool):
设置参数的 trainable 属性
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
get
\_
prediction (bool): 是否执行预测。
...
...
hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/README.md
浏览文件 @
23575238
```
shell
$
hub
install
yolov3_darknet53_coco2017
==
1.1.0
```
## 命令行预测
```
hub run yolov3_darknet53_coco2017 --input_path "/PATH/TO/IMAGE"
$
hub run yolov3_darknet53_coco2017 --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -16,11 +12,11 @@ def context(trainable=True,
get_prediction=False)
```
特征提取
,用于迁移学习。
提取特征
,用于迁移学习。
**参数**
*
trainable(bool):
将参数的trainable属性设为trainable
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
get
\_
prediction (bool): 是否执行预测。
...
...
@@ -29,7 +25,7 @@ def context(trainable=True,
*
inputs (dict): 模型的输入,keys 包括 'image', 'im
\_
size',相应的取值为:
*
image (Variable): 图像变量
*
im
\_
size (Variable): 图片的尺寸
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
fatures',否则输出 'bbox
\_
out'。
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
f
eatures'、'body
\_
fe
atures',否则输出 'bbox
\_
out'。
*
context
\_
prog (Program): 用于迁移学习的 Program.
```
python
...
...
hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/module.py
浏览文件 @
23575238
...
...
@@ -88,7 +88,7 @@ class YOLOv3DarkNet53Coco2017(hub.Module):
# yolo_head
yolo_head
=
YOLOv3Head
(
num_classes
=
80
)
# head_features
head_features
=
yolo_head
.
_get_outputs
(
head_features
,
body_features
=
yolo_head
.
_get_outputs
(
body_feats
,
is_train
=
trainable
)
place
=
fluid
.
CPUPlace
()
...
...
@@ -109,7 +109,9 @@ class YOLOv3DarkNet53Coco2017(hub.Module):
else
:
outputs
=
{
'head_features'
:
[
var_prefix
+
var
.
name
for
var
in
head_features
]
[
var_prefix
+
var
.
name
for
var
in
head_features
],
'body_features'
:
[
var_prefix
+
var
.
name
for
var
in
body_features
]
}
# add_vars_prefix
add_vars_prefix
(
context_prog
,
var_prefix
)
...
...
hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/yolo_head.py
浏览文件 @
23575238
...
...
@@ -227,7 +227,7 @@ class YOLOv3Head(object):
# upsample
route
=
self
.
_upsample
(
route
)
return
outputs
return
outputs
,
blocks
def
get_prediction
(
self
,
outputs
,
im_size
):
"""
...
...
hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/README.md
浏览文件 @
23575238
```
shell
$
hub
install
yolov3_darknet53_pedestrian
==
1.0.0
```
## 命令行预测
```
hub run yolov3_darknet53_pedestrian --input_path "/PATH/TO/IMAGE"
$
hub run yolov3_darknet53_pedestrian --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -16,11 +12,11 @@ def context(trainable=True,
get_prediction=False)
```
特征提取
,用于迁移学习。
提取特征
,用于迁移学习。
**参数**
*
trainable(bool):
将参数的trainable属性设为trainable
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
get
\_
prediction (bool): 是否执行预测。
...
...
@@ -29,7 +25,7 @@ def context(trainable=True,
*
inputs (dict): 模型的输入,keys 包括 'image', 'im
\_
size',相应的取值为:
*
image (Variable): 图像变量
*
im
\_
size (Variable): 图片的尺寸
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
fatures',否则输出 'bbox
\_
out'。
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
f
eatures'、'body
\_
fe
atures',否则输出 'bbox
\_
out'。
*
context
\_
prog (Program): 用于迁移学习的 Program.
```
python
...
...
hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/module.py
浏览文件 @
23575238
...
...
@@ -102,7 +102,7 @@ class YOLOv3DarkNet53Pedestrian(hub.Module):
normalized
=
False
,
score_threshold
=
0.01
))
# head_features
head_features
=
yolo_head
.
_get_outputs
(
head_features
,
body_features
=
yolo_head
.
_get_outputs
(
body_feats
,
is_train
=
trainable
)
place
=
fluid
.
CPUPlace
()
...
...
@@ -123,7 +123,9 @@ class YOLOv3DarkNet53Pedestrian(hub.Module):
else
:
outputs
=
{
'head_features'
:
[
var_prefix
+
var
.
name
for
var
in
head_features
]
[
var_prefix
+
var
.
name
for
var
in
head_features
],
'body_features'
:
[
var_prefix
+
var
.
name
for
var
in
body_features
]
}
# add_vars_prefix
add_vars_prefix
(
context_prog
,
var_prefix
)
...
...
hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/yolo_head.py
浏览文件 @
23575238
...
...
@@ -227,7 +227,7 @@ class YOLOv3Head(object):
# upsample
route
=
self
.
_upsample
(
route
)
return
outputs
return
outputs
,
blocks
def
get_prediction
(
self
,
outputs
,
im_size
):
"""
...
...
hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/README.md
浏览文件 @
23575238
```
shell
$
hub
install
yolov3_darknet53_vehicles
==
1.0.0
```
## 命令行预测
```
hub run yolov3_darknet53_vehicles --input_path "/PATH/TO/IMAGE"
$
hub run yolov3_darknet53_vehicles --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -16,11 +12,11 @@ def context(trainable=True,
get_prediction=False)
```
特征提取
,用于迁移学习。
提取特征
,用于迁移学习。
**参数**
*
trainable(bool):
将参数的trainable属性设为trainable
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
get
\_
prediction (bool): 是否执行预测。
...
...
@@ -29,7 +25,7 @@ def context(trainable=True,
*
inputs (dict): 模型的输入,keys 包括 'image', 'im
\_
size',相应的取值为:
*
image (Variable): 图像变量
*
im
\_
size (Variable): 图片的尺寸
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
fatures',否则输出 'bbox
\_
out'。
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
f
eatures'、'body
\_
fe
atures',否则输出 'bbox
\_
out'。
*
context
\_
prog (Program): 用于迁移学习的 Program.
```
python
...
...
hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/module.py
浏览文件 @
23575238
...
...
@@ -102,7 +102,7 @@ class YOLOv3DarkNet53Vehicles(hub.Module):
normalized
=
False
,
score_threshold
=
0.005
))
# head_features
head_features
=
yolo_head
.
_get_outputs
(
head_features
,
body_features
=
yolo_head
.
_get_outputs
(
body_feats
,
is_train
=
trainable
)
place
=
fluid
.
CPUPlace
()
...
...
@@ -123,7 +123,9 @@ class YOLOv3DarkNet53Vehicles(hub.Module):
else
:
outputs
=
{
'head_features'
:
[
var_prefix
+
var
.
name
for
var
in
head_features
]
[
var_prefix
+
var
.
name
for
var
in
head_features
],
'body_features'
:
[
var_prefix
+
var
.
name
for
var
in
body_features
]
}
# add_vars_prefix
add_vars_prefix
(
context_prog
,
var_prefix
)
...
...
hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/yolo_head.py
浏览文件 @
23575238
...
...
@@ -227,7 +227,7 @@ class YOLOv3Head(object):
# upsample
route
=
self
.
_upsample
(
route
)
return
outputs
return
outputs
,
blocks
def
get_prediction
(
self
,
outputs
,
im_size
):
"""
...
...
hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/README.md
浏览文件 @
23575238
```
shell
hub
install
yolov3_mobilenet_v1_coco2017
==
1.1.0
```
## 命令行预测
```
hub run yolov3_mobilenet_v1_coco2017 --input_path "/PATH/TO/IMAGE"
$
hub run yolov3_mobilenet_v1_coco2017 --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -16,11 +12,11 @@ def context(trainable=True,
get_prediction=False)
```
特征提取
,用于迁移学习。
提取特征
,用于迁移学习。
**参数**
*
trainable(bool):
将参数的 trainable 属性设为 trainable
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
get
\_
prediction (bool): 是否执行预测。
...
...
@@ -29,7 +25,7 @@ def context(trainable=True,
*
inputs (dict): 模型的输入,keys 包括 'image', 'im
\_
size',相应的取值为:
*
image (Variable): 图像变量;
*
im
\_
size (Variable): 图片的尺寸
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
fatures',否则输出 'bbox
\_
out'。
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
f
eatures'、'body
\_
fe
atures',否则输出 'bbox
\_
out'。
*
context
\_
prog (Program): 用于迁移学习的 Program.
```
python
...
...
hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/module.py
浏览文件 @
23575238
...
...
@@ -92,7 +92,7 @@ class YOLOv3MobileNetV1Coco2017(hub.Module):
# yolo_head
yolo_head
=
YOLOv3Head
(
num_classes
=
80
)
# head_features
head_features
=
yolo_head
.
_get_outputs
(
head_features
,
body_features
=
yolo_head
.
_get_outputs
(
body_feats
,
is_train
=
trainable
)
place
=
fluid
.
CPUPlace
()
...
...
@@ -113,7 +113,9 @@ class YOLOv3MobileNetV1Coco2017(hub.Module):
else
:
outputs
=
{
'head_features'
:
[
var_prefix
+
var
.
name
for
var
in
head_features
]
[
var_prefix
+
var
.
name
for
var
in
head_features
],
'body_features'
:
[
var_prefix
+
var
.
name
for
var
in
body_features
]
}
# add_vars_prefix
add_vars_prefix
(
context_prog
,
var_prefix
)
...
...
hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/yolo_head.py
浏览文件 @
23575238
...
...
@@ -227,7 +227,7 @@ class YOLOv3Head(object):
# upsample
route
=
self
.
_upsample
(
route
)
return
outputs
return
outputs
,
blocks
def
get_prediction
(
self
,
outputs
,
im_size
):
"""
...
...
hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/README.md
浏览文件 @
23575238
```
shell
$
hub
install
yolov3_resnet34_coco2017
==
1.1.0
```
## 命令行预测
```
hub run yolov3_resnet34_coco2017 --input_path "/PATH/TO/IMAGE"
$
hub run yolov3_resnet34_coco2017 --input_path "/PATH/TO/IMAGE"
```
## API
...
...
@@ -16,11 +12,11 @@ def context(trainable=True,
get_prediction=False)
```
提取
头部
特征,用于迁移学习。
提取特征,用于迁移学习。
**参数**
*
trainable(bool):
将参数的trainable属性设为trainable
;
*
trainable(bool):
参数是否可训练
;
*
pretrained (bool): 是否加载预训练模型;
*
get
\_
prediction (bool): 是否执行预测。
...
...
@@ -29,7 +25,7 @@ def context(trainable=True,
*
inputs (dict): 模型的输入,keys 包括 'image', 'im
\_
size',相应的取值为:
*
image (Variable): 图像变量
*
im
\_
size (Variable): 图片的尺寸
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
fatures',否则输出 'bbox
\_
out'。
*
outputs (dict): 模型的输出。如果 get
\_
prediction 为 False,输出 'head
\_
f
eatures'、'body
\_
fe
atures',否则输出 'bbox
\_
out'。
*
context
\_
prog (Program): 用于迁移学习的 Program.
```
python
...
...
hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py
浏览文件 @
23575238
...
...
@@ -94,7 +94,7 @@ class YOLOv3DarkNet53Coco2017(hub.Module):
# yolo_head
yolo_head
=
YOLOv3Head
(
num_classes
=
80
)
# head_features
head_features
=
yolo_head
.
_get_outputs
(
head_features
,
body_features
=
yolo_head
.
_get_outputs
(
body_feats
,
is_train
=
trainable
)
place
=
fluid
.
CPUPlace
()
...
...
@@ -115,7 +115,9 @@ class YOLOv3DarkNet53Coco2017(hub.Module):
else
:
outputs
=
{
'head_features'
:
[
var_prefix
+
var
.
name
for
var
in
head_features
]
[
var_prefix
+
var
.
name
for
var
in
head_features
],
'body_features'
:
[
var_prefix
+
var
.
name
for
var
in
body_features
]
}
# add_vars_prefix
add_vars_prefix
(
context_prog
,
var_prefix
)
...
...
hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/yolo_head.py
浏览文件 @
23575238
...
...
@@ -227,7 +227,7 @@ class YOLOv3Head(object):
# upsample
route
=
self
.
_upsample
(
route
)
return
outputs
return
outputs
,
blocks
def
get_prediction
(
self
,
outputs
,
im_size
):
"""
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录