Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleDetection
提交
83714347
P
PaddleDetection
项目概览
PaddlePaddle
/
PaddleDetection
大约 1 年 前同步成功
通知
695
Star
11112
Fork
2696
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
184
列表
看板
标记
里程碑
合并请求
40
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
184
Issue
184
列表
看板
标记
里程碑
合并请求
40
合并请求
40
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
83714347
编写于
3月 22, 2021
作者:
K
Kaipeng Deng
提交者:
GitHub
3月 22, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
polish weights (#2376)
* unity weights
上级
abe1ffe4
变更
39
显示空白变更内容
内联
并排
Showing
39 changed file
with
74 addition
and
81 deletion
+74
-81
dygraph/configs/dcn/cascade_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml
...onfigs/dcn/cascade_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml
+1
-1
dygraph/configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml
dygraph/configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml
+1
-1
dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_1x_coco.yml
dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_1x_coco.yml
+1
-1
dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml
dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml
+1
-1
dygraph/configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml
...configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml
+1
-1
dygraph/configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml
dygraph/configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml
+1
-1
dygraph/configs/dcn/mask_rcnn_dcn_r50_vd_fpn_2x_coco.yml
dygraph/configs/dcn/mask_rcnn_dcn_r50_vd_fpn_2x_coco.yml
+1
-1
dygraph/configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml
...h/configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml
dygraph/configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml
dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.yml
dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml
...h/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml
...h/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml
dygraph/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml
...ph/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml
dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml
...ph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_2x_coco.yml
...ph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_2x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml
...igs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml
+1
-1
dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_2x_coco.yml
...igs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_2x_coco.yml
+1
-1
dygraph/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.yml
dygraph/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.yml
+1
-1
dygraph/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml
dygraph/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml
+1
-1
dygraph/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco.yml
dygraph/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco.yml
+1
-1
dygraph/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_2x_coco.yml
dygraph/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_2x_coco.yml
+1
-1
dygraph/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml
...configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml
+1
-1
dygraph/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_2x_coco.yml
...configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_2x_coco.yml
+1
-1
dygraph/configs/slim/prune/yolov3_prune_fpgm.yml
dygraph/configs/slim/prune/yolov3_prune_fpgm.yml
+1
-1
dygraph/configs/slim/prune/yolov3_prune_l1_norm.yml
dygraph/configs/slim/prune/yolov3_prune_l1_norm.yml
+1
-1
dygraph/configs/slim/quant/yolov3_mobilenet_v1_qat.yml
dygraph/configs/slim/quant/yolov3_mobilenet_v1_qat.yml
+1
-1
dygraph/configs/slim/quant/yolov3_mobilenet_v3_qat.yml
dygraph/configs/slim/quant/yolov3_mobilenet_v3_qat.yml
+1
-1
dygraph/configs/ssd/_base_/ssd_vgg16_300.yml
dygraph/configs/ssd/_base_/ssd_vgg16_300.yml
+1
-1
dygraph/configs/yolov3/yolov3_mobilenet_v1_roadsign.yml
dygraph/configs/yolov3/yolov3_mobilenet_v1_roadsign.yml
+1
-1
dygraph/ppdet/engine/trainer.py
dygraph/ppdet/engine/trainer.py
+9
-12
dygraph/ppdet/utils/checkpoint.py
dygraph/ppdet/utils/checkpoint.py
+23
-23
dygraph/ppdet/utils/download.py
dygraph/ppdet/utils/download.py
+1
-1
dygraph/tools/eval.py
dygraph/tools/eval.py
+1
-1
dygraph/tools/export_model.py
dygraph/tools/export_model.py
+1
-1
dygraph/tools/infer.py
dygraph/tools/infer.py
+1
-1
dygraph/tools/train.py
dygraph/tools/train.py
+6
-10
未找到文件。
dygraph/configs/dcn/cascade_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
cascade_rcnn_dcn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNeXt101_vd_64x4d_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams
weights
:
output/cascade_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
faster_rcnn_dcn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet101_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams
weights
:
output/faster_rcnn_dcn_r101_vd_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
faster_rcnn_dcn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams
weights
:
output/faster_rcnn_dcn_r50_vd_fpn_2x_coco/model_final
ResNet
:
...
...
dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
faster_rcnn_dcn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams
weights
:
output/faster_rcnn_dcn_r50_vd_fpn_2x_coco/model_final
ResNet
:
...
...
dygraph/configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
faster_rcnn_dcn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNeXt101_vd_64x4d_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams
weights
:
output/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
mask_rcnn_dcn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet101_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams
weights
:
output/mask_rcnn_dcn_r101_vd_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/dcn/mask_rcnn_dcn_r50_vd_fpn_2x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
mask_rcnn_dcn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams
weights
:
output/mask_rcnn_dcn_r50_vd_fpn_2x_coco/model_final
ResNet
:
...
...
dygraph/configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
mask_rcnn_dcn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNeXt101_vd_64x4d_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams
weights
:
output/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
faster_rcnn_r50_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet101_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams
weights
:
output/faster_rcnn_r101_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet101_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams
weights
:
output/faster_rcnn_r101_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet101_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams
weights
:
output/faster_rcnn_r101_fpn_2x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet101_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams
weights
:
output/faster_rcnn_r101_vd_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet101_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams
weights
:
output/faster_rcnn_r101_vd_fpn_2x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet34_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet34_pretrained.pdparams
weights
:
output/faster_rcnn_r34_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet34_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet34_vd_pretrained.pdparams
weights
:
output/faster_rcnn_r34_vd_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
faster_rcnn_r50_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams
weights
:
output/faster_rcnn_r50_vd_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams
weights
:
output/faster_rcnn_r50_vd_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_2x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams
weights
:
output/faster_rcnn_r50_vd_fpn_2x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNeXt101_vd_64x4d_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams
weights
:
output/faster_rcnn_x101_vd_64x4d_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_2x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
faster_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNeXt101_vd_64x4d_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams
weights
:
output/faster_rcnn_x101_vd_64x4d_fpn_2x_coco/model_final
ResNet
:
...
...
dygraph/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
mask_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet101_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams
weights
:
output/mask_rcnn_r101_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml
浏览文件 @
83714347
_BASE_
:
[
'
mask_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet101_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams
weights
:
output/mask_rcnn_r101_vd_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
mask_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams
weights
:
output/mask_rcnn_r50_vd_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_2x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
mask_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams
weights
:
output/mask_rcnn_r50_vd_fpn_2x_coco/model_final
ResNet
:
...
...
dygraph/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
mask_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNeXt101_vd_64x4d_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams
weights
:
output/mask_rcnn_x101_vd_64x4d_fpn_1x_coco/model_final
ResNet
:
...
...
dygraph/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_2x_coco.yml
浏览文件 @
83714347
...
...
@@ -2,7 +2,7 @@ _BASE_: [
'
mask_rcnn_r50_fpn_1x_coco.yml'
,
]
pretrain_weights
:
https://paddle
-imagenet-models-name.bj.bcebos.com/ResNeXt101_vd_64x4d_pretrained.tar
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams
weights
:
output/mask_rcnn_x101_vd_64x4d_fpn_2x_coco/model_final
ResNet
:
...
...
dygraph/configs/slim/prune/yolov3_prune_fpgm.yml
浏览文件 @
83714347
# Weights of yolov3_mobilenet_v1_voc
pretrain_weights
:
https://paddle
models.bj.bcebos.com/object_detection/dygraph
/yolov3_mobilenet_v1_270e_voc.pdparams
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models
/yolov3_mobilenet_v1_270e_voc.pdparams
weight_type
:
resume
slim
:
Pruner
...
...
dygraph/configs/slim/prune/yolov3_prune_l1_norm.yml
浏览文件 @
83714347
# Weights of yolov3_mobilenet_v1_voc
pretrain_weights
:
https://paddle
models.bj.bcebos.com/object_detection/dygraph
/yolov3_mobilenet_v1_270e_voc.pdparams
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models
/yolov3_mobilenet_v1_270e_voc.pdparams
weight_type
:
resume
slim
:
Pruner
...
...
dygraph/configs/slim/quant/yolov3_mobilenet_v1_qat.yml
浏览文件 @
83714347
# Weights of yolov3_mobilenet_v1_coco
pretrain_weights
:
https://paddle
models.bj.bcebos.com/object_detection/dygraph
/yolov3_mobilenet_v1_270e_coco.pdparams
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models
/yolov3_mobilenet_v1_270e_coco.pdparams
weight_type
:
resume
slim
:
QAT
...
...
dygraph/configs/slim/quant/yolov3_mobilenet_v3_qat.yml
浏览文件 @
83714347
# Weights of yolov3_mobilenet_v3_coco
pretrain_weights
:
https://paddle
models.bj.bcebos.com/object_detection/dygraph
/yolov3_mobilenet_v3_large_270e_coco.pdparams
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models
/yolov3_mobilenet_v3_large_270e_coco.pdparams
weight_type
:
resume
slim
:
QAT
...
...
dygraph/configs/ssd/_base_/ssd_vgg16_300.yml
浏览文件 @
83714347
architecture
:
SSD
pretrain_weights
:
https://paddle
models.bj.bcebos.com/object_detection/dygraph
/VGG16_caffe_pretrained.pdparams
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models/pretrained
/VGG16_caffe_pretrained.pdparams
# Model Achitecture
SSD
:
...
...
dygraph/configs/yolov3/yolov3_mobilenet_v1_roadsign.yml
浏览文件 @
83714347
...
...
@@ -4,7 +4,7 @@ _BASE_: [
'
_base_/yolov3_mobilenet_v1.yml'
,
'
_base_/yolov3_reader.yml'
,
]
pretrain_weights
:
https://paddle
models.bj.bcebos.com/object_detection/dygraph
/yolov3_mobilenet_v1_270e_coco.pdparams
pretrain_weights
:
https://paddle
det.bj.bcebos.com/models
/yolov3_mobilenet_v1_270e_coco.pdparams
norm_type
:
sync_bn
weights
:
output/yolov3_mobilenet_v1_roadsign/model_final
metric
:
VOC
...
...
dygraph/ppdet/engine/trainer.py
浏览文件 @
83714347
...
...
@@ -58,7 +58,7 @@ class Trainer(object):
# model slim build
if
'slim'
in
cfg
and
cfg
.
slim
:
if
self
.
mode
==
'train'
:
self
.
load_weights
(
cfg
.
pretrain_weights
,
cfg
.
weight_type
)
self
.
load_weights
(
cfg
.
pretrain_weights
)
self
.
slim
=
create
(
cfg
.
slim
)
self
.
slim
(
self
.
model
)
...
...
@@ -174,17 +174,14 @@ class Trainer(object):
"metrics shoule be instances of subclass of Metric"
self
.
_metrics
.
extend
(
metrics
)
def
load_weights
(
self
,
weights
,
weight_type
=
'pretrain'
):
assert
weight_type
in
[
'pretrain'
,
'resume'
,
'finetune'
],
\
"weight_type can only be 'pretrain', 'resume', 'finetune'"
if
weight_type
==
'resume'
:
def
load_weights
(
self
,
weights
):
self
.
start_epoch
=
0
load_pretrain_weight
(
self
.
model
,
weights
)
logger
.
debug
(
"Load weights {} to start training"
.
format
(
weights
))
def
resume_weights
(
self
,
weights
):
self
.
start_epoch
=
load_weight
(
self
.
model
,
weights
,
self
.
optimizer
)
logger
.
debug
(
"Resume weights of epoch {}"
.
format
(
self
.
start_epoch
))
else
:
self
.
start_epoch
=
0
load_pretrain_weight
(
self
.
model
,
weights
,
weight_type
)
logger
.
debug
(
"Load {} weights {} to start training"
.
format
(
weight_type
,
weights
))
def
train
(
self
,
validate
=
False
):
assert
self
.
mode
==
'train'
,
"Model not in 'train' mode"
...
...
dygraph/ppdet/utils/checkpoint.py
浏览文件 @
83714347
...
...
@@ -123,8 +123,7 @@ def load_weight(model, weight, optimizer=None):
assert
incorrect_keys
==
0
,
"Load weight {} incorrectly,
\
{} keys unmatched, please check again."
.
format
(
weight
,
incorrect_keys
)
logger
.
info
(
'Finish loading model weight parameter: {}'
.
format
(
pdparam_path
))
logger
.
info
(
'Finish resuming model weights: {}'
.
format
(
pdparam_path
))
model
.
set_dict
(
model_weight
)
...
...
@@ -142,9 +141,7 @@ def load_weight(model, weight, optimizer=None):
return
last_epoch
def
load_pretrain_weight
(
model
,
pretrain_weight
,
weight_type
=
'pretrain'
):
assert
weight_type
in
[
'pretrain'
,
'finetune'
]
def
load_pretrain_weight
(
model
,
pretrain_weight
):
if
is_url
(
pretrain_weight
):
pretrain_weight
=
get_weights_path_dist
(
pretrain_weight
)
...
...
@@ -158,24 +155,27 @@ def load_pretrain_weight(model, pretrain_weight, weight_type='pretrain'):
model_dict
=
model
.
state_dict
()
param_state_dict
=
paddle
.
load
(
path
+
'.pdparams'
)
if
weight_type
==
'pretrain'
:
model
.
backbone
.
set_dict
(
param_state_dict
)
else
:
weights_path
=
path
+
'.pdparams'
param_state_dict
=
paddle
.
load
(
weights_path
)
ignore_set
=
set
()
lack_modules
=
set
()
for
name
,
weight
in
model_dict
.
items
():
if
name
in
param_state_dict
.
keys
():
if
weight
.
shape
!=
list
(
param_state_dict
[
name
].
shape
):
logger
.
info
(
'{} not used, shape {} unmatched with {} in model.'
.
format
(
name
,
list
(
param_state_dict
[
name
].
shape
),
weight
.
shape
))
'{} not used, shape {} unmatched with {} in model.'
.
format
(
name
,
list
(
param_state_dict
[
name
].
shape
),
weight
.
shape
))
param_state_dict
.
pop
(
name
,
None
)
else
:
logger
.
info
(
'Lack weight: {}'
.
format
(
name
))
lack_modules
.
add
(
name
.
split
(
'.'
)[
0
])
logger
.
debug
(
'Lack weights: {}'
.
format
(
name
))
if
len
(
lack_modules
)
>
0
:
logger
.
info
(
'Lack weights of modules: {}'
.
format
(
', '
.
join
(
list
(
lack_modules
))))
model
.
set_dict
(
param_state_dict
)
return
logger
.
info
(
'Finish loading model weights: {}'
.
format
(
weights_path
))
def
save_model
(
model
,
optimizer
,
save_dir
,
save_name
,
last_epoch
):
...
...
dygraph/ppdet/utils/download.py
浏览文件 @
83714347
...
...
@@ -92,7 +92,7 @@ DATASETS = {
DOWNLOAD_RETRY_LIMIT
=
3
PPDET_WEIGHTS_DOWNLOAD_URL_PREFIX
=
'https://paddle
models.bj.bcebos.com/object_detection
/'
PPDET_WEIGHTS_DOWNLOAD_URL_PREFIX
=
'https://paddle
det.bj.bcebos.com
/'
def
parse_url
(
url
):
...
...
dygraph/tools/eval.py
浏览文件 @
83714347
...
...
@@ -93,7 +93,7 @@ def run(FLAGS, cfg):
trainer
=
Trainer
(
cfg
,
mode
=
'eval'
)
# load weights
trainer
.
load_weights
(
cfg
.
weights
,
'resume'
)
trainer
.
load_weights
(
cfg
.
weights
)
# training
trainer
.
evaluate
()
...
...
dygraph/tools/export_model.py
浏览文件 @
83714347
...
...
@@ -62,7 +62,7 @@ def run(FLAGS, cfg):
trainer
=
Trainer
(
cfg
,
mode
=
'test'
)
# load weights
trainer
.
load_weights
(
cfg
.
weights
,
'resume'
)
trainer
.
load_weights
(
cfg
.
weights
)
# export model
trainer
.
export
(
FLAGS
.
output_dir
)
...
...
dygraph/tools/infer.py
浏览文件 @
83714347
...
...
@@ -114,7 +114,7 @@ def run(FLAGS, cfg):
trainer
=
Trainer
(
cfg
,
mode
=
'test'
)
# load weights
trainer
.
load_weights
(
cfg
.
weights
,
'resume'
)
trainer
.
load_weights
(
cfg
.
weights
)
# get inference images
images
=
get_test_images
(
FLAGS
.
infer_dir
,
FLAGS
.
infer_img
)
...
...
dygraph/tools/train.py
浏览文件 @
83714347
...
...
@@ -43,17 +43,13 @@ logger = setup_logger('train')
def
parse_args
():
parser
=
cli
.
ArgsParser
()
parser
.
add_argument
(
"--weight_type"
,
default
=
'pretrain'
,
type
=
str
,
help
=
"Loading Checkpoints only support 'pretrain', 'finetune', 'resume'."
)
parser
.
add_argument
(
"--eval"
,
action
=
'store_true'
,
default
=
False
,
help
=
"Whether to perform evaluation in train"
)
parser
.
add_argument
(
"-r"
,
"--resume"
,
default
=
None
,
help
=
"weights path for resume"
)
parser
.
add_argument
(
"--slim_config"
,
default
=
None
,
...
...
@@ -101,8 +97,10 @@ def run(FLAGS, cfg):
trainer
=
Trainer
(
cfg
,
mode
=
'train'
)
# load weights
if
not
FLAGS
.
slim_config
and
'pretrain_weights'
in
cfg
and
cfg
.
pretrain_weights
:
trainer
.
load_weights
(
cfg
.
pretrain_weights
,
FLAGS
.
weight_type
)
if
FLAGS
.
resume
is
not
None
:
trainer
.
resume_weights
(
FLAGS
.
resume
)
elif
not
FLAGS
.
slim_config
and
'pretrain_weights'
in
cfg
and
cfg
.
pretrain_weights
:
trainer
.
load_weights
(
cfg
.
pretrain_weights
)
# training
trainer
.
train
(
FLAGS
.
eval
)
...
...
@@ -120,8 +118,6 @@ def main():
if
FLAGS
.
slim_config
:
slim_cfg
=
load_config
(
FLAGS
.
slim_config
)
merge_config
(
slim_cfg
)
if
'weight_type'
not
in
cfg
:
cfg
.
weight_type
=
FLAGS
.
weight_type
check
.
check_config
(
cfg
)
check
.
check_gpu
(
cfg
.
use_gpu
)
check
.
check_version
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录