Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleX
提交
6c0de407
P
PaddleX
项目概览
PaddlePaddle
/
PaddleX
通知
136
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
6c0de407
编写于
6月 05, 2020
作者:
F
FlyingQianMM
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
use ComposedRCNNTransforms in negatives training example
上级
bec64c22
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
45 addition
and
12 deletion
+45
-12
docs/tuning_strategy/detection/negatives_training.md
docs/tuning_strategy/detection/negatives_training.md
+6
-12
tutorials/train/detection/test.py
tutorials/train/detection/test.py
+39
-0
未找到文件。
docs/tuning_strategy/detection/negatives_training.md
浏览文件 @
6c0de407
...
...
@@ -41,17 +41,10 @@ from paddlex.det import transforms
import
paddlex
as
pdx
#
定义训练和验证时的
transforms
train_transforms
=
transforms
.
Compose
([
transforms
.
RandomHorizontalFlip
(),
transforms
.
Normalize
(),
transforms
.
ResizeByShort
(
short_size
=
600
,
max_size
=
1000
),
transforms
.
Padding
(
coarsest_stride
=
32
)
])
eval_transforms
=
transforms
.
Compose
([
transforms
.
Normalize
(),
transforms
.
ResizeByShort
(
short_size
=
600
,
max_size
=
1000
),
transforms
.
Padding
(
coarsest_stride
=
32
),
])
train_transforms
=
transforms
.
ComposedRCNNTransforms
(
mode
=
'train'
,
min_max_size
=[
600
,
1000
])
eval_transforms
=
transforms
.
ComposedRCNNTransforms
(
mode
=
'eval'
,
min_max_size
=[
600
,
1000
])
#
定义训练所用的数据集
train_dataset
=
pdx
.
datasets
.
CocoDetection
(
...
...
@@ -61,7 +54,8 @@ train_dataset = pdx.datasets.CocoDetection(
shuffle
=
True
,
num_workers
=
2
)
#
训练集中加入无目标背景图片
train_dataset
.
add_negative_samples
(
'jinnan2_round1_train_20190305/normal_train_back/'
)
train_dataset
.
add_negative_samples
(
'jinnan2_round1_train_20190305/normal_train_back/'
)
#
定义验证所用的数据集
eval_dataset
=
pdx
.
datasets
.
CocoDetection
(
...
...
tutorials/train/detection/test.py
0 → 100644
浏览文件 @
6c0de407
import
os
os
.
environ
[
'CUDA_VISIBLE_DEVICES'
]
=
'0'
from
paddlex.det
import
transforms
import
paddlex
as
pdx
# 定义训练和验证时的transforms
train_transforms
=
transforms
.
ComposedRCNNTransforms
(
mode
=
'train'
,
min_max_size
=
[
600
,
1000
])
eval_transforms
=
transforms
.
ComposedRCNNTransforms
(
mode
=
'eval'
,
min_max_size
=
[
600
,
1000
])
# 定义训练所用的数据集
train_dataset
=
pdx
.
datasets
.
CocoDetection
(
data_dir
=
'jinnan2_round1_train_20190305/restricted/'
,
ann_file
=
'jinnan2_round1_train_20190305/train.json'
,
transforms
=
train_transforms
,
shuffle
=
True
,
num_workers
=
2
)
# 训练集中加入无目标背景图片
train_dataset
.
add_negative_samples
(
'jinnan2_round1_train_20190305/normal_train_back/'
)
# 定义验证所用的数据集
eval_dataset
=
pdx
.
datasets
.
CocoDetection
(
data_dir
=
'jinnan2_round1_train_20190305/restricted/'
,
ann_file
=
'jinnan2_round1_train_20190305/val.json'
,
transforms
=
eval_transforms
,
num_workers
=
2
)
# 初始化模型,并进行训练
model
=
pdx
.
det
.
FasterRCNN
(
num_classes
=
len
(
train_dataset
.
labels
)
+
1
)
model
.
train
(
num_epochs
=
17
,
train_dataset
=
train_dataset
,
eval_dataset
=
eval_dataset
,
train_batch_size
=
8
,
learning_rate
=
0.01
,
lr_decay_epochs
=
[
13
,
16
],
save_dir
=
'./output'
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录