Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
6a569885
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
289
Star
8
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSeg
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
6a569885
编写于
6月 17, 2020
作者:
C
chenguowei01
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update dataset using
上级
78b49980
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
44 addition
and
61 deletion
+44
-61
dygraph/infer.py
dygraph/infer.py
+13
-17
dygraph/train.py
dygraph/train.py
+17
-26
dygraph/val.py
dygraph/val.py
+14
-18
未找到文件。
dygraph/infer.py
浏览文件 @
6a569885
...
...
@@ -43,22 +43,11 @@ def parse_args():
# params of dataset
parser
.
add_argument
(
'--data_dir'
,
dest
=
'data_dir'
,
help
=
'The root directory of dataset'
,
type
=
str
)
parser
.
add_argument
(
'--test_list'
,
dest
=
'test_list'
,
help
=
'Val list file of dataset'
,
'--dataset'
,
dest
=
'dataset'
,
help
=
'The dataset you want to train'
,
type
=
str
,
default
=
None
)
parser
.
add_argument
(
'--num_classes'
,
dest
=
'num_classes'
,
help
=
'Number of classes'
,
type
=
int
,
default
=
2
)
default
=
'OpticDiscSeg'
)
# params of prediction
parser
.
add_argument
(
...
...
@@ -142,12 +131,19 @@ def main(args):
places
=
fluid
.
CUDAPlace
(
ParallelEnv
().
dev_id
)
\
if
env_info
[
'place'
]
==
'cuda'
and
fluid
.
is_compiled_with_cuda
()
\
else
fluid
.
CPUPlace
()
if
args
.
dataset
.
lower
()
==
'opticdiscseg'
:
dataset
=
OpticDiscSeg
else
:
raise
Exception
(
"The --dataset set wrong. It should be one of ('OpticDiscSeg',)"
)
with
fluid
.
dygraph
.
guard
(
places
):
test_transforms
=
T
.
Compose
([
T
.
Resize
(
args
.
input_size
),
T
.
Normalize
()])
test_dataset
=
OpticDiscSeg
(
transforms
=
test_transforms
,
mode
=
'test'
)
test_dataset
=
dataset
(
transforms
=
test_transforms
,
mode
=
'test'
)
if
args
.
model_name
==
'UNet'
:
model
=
models
.
UNet
(
num_classes
=
args
.
num_classes
)
model
=
models
.
UNet
(
num_classes
=
test_dataset
.
num_classes
)
infer
(
model
,
...
...
dygraph/train.py
浏览文件 @
6a569885
...
...
@@ -40,31 +40,15 @@ def parse_args():
dest
=
'model_name'
,
help
=
"Model type for traing, which is one of ('UNet')"
,
type
=
str
,
default
=
'
UNet
'
)
default
=
'
OpticDiscSeg
'
)
# params of dataset
parser
.
add_argument
(
'--data_dir'
,
dest
=
'data_dir'
,
help
=
'The root directory of dataset'
,
type
=
str
)
parser
.
add_argument
(
'--train_list'
,
dest
=
'train_list'
,
help
=
'Train list file of dataset'
,
type
=
str
)
parser
.
add_argument
(
'--val_list'
,
dest
=
'val_list'
,
help
=
'Val list file of dataset'
,
'--dataset'
,
dest
=
'dataset'
,
help
=
'The dataset you want to train'
,
type
=
str
,
default
=
None
)
parser
.
add_argument
(
'--num_classes'
,
dest
=
'num_classes'
,
help
=
'Number of classes'
,
type
=
int
,
default
=
2
)
default
=
'OpticDiscSeg'
)
# params of training
parser
.
add_argument
(
...
...
@@ -83,7 +67,7 @@ def parse_args():
parser
.
add_argument
(
'--batch_size'
,
dest
=
'batch_size'
,
help
=
'Mini batch size'
,
help
=
'Mini batch size
of one gpu or cpu
'
,
type
=
int
,
default
=
2
)
parser
.
add_argument
(
...
...
@@ -210,6 +194,12 @@ def main(args):
if
env_info
[
'place'
]
==
'cuda'
and
fluid
.
is_compiled_with_cuda
()
\
else
fluid
.
CPUPlace
()
if
args
.
dataset
.
lower
()
==
'opticdiscseg'
:
dataset
=
OpticDiscSeg
else
:
raise
Exception
(
"The --dataset set wrong. It should be one of ('OpticDiscSeg',)"
)
with
fluid
.
dygraph
.
guard
(
places
):
# Creat dataset reader
train_transforms
=
T
.
Compose
([
...
...
@@ -217,17 +207,18 @@ def main(args):
T
.
RandomHorizontalFlip
(),
T
.
Normalize
()
])
train_dataset
=
OpticDiscSeg
(
transforms
=
train_transforms
,
mode
=
'train'
)
train_dataset
=
dataset
(
transforms
=
train_transforms
,
mode
=
'train'
)
eval_dataset
=
None
if
args
.
do_eval
:
eval_transforms
=
T
.
Compose
(
[
T
.
Resize
(
args
.
input_size
),
T
.
Normalize
()])
eval_dataset
=
OpticDiscSeg
(
transforms
=
eval_transforms
,
mode
=
'eval'
)
eval_dataset
=
dataset
(
transforms
=
eval_transforms
,
mode
=
'eval'
)
if
args
.
model_name
==
'UNet'
:
model
=
models
.
UNet
(
num_classes
=
args
.
num_classes
,
ignore_index
=
255
)
model
=
models
.
UNet
(
num_classes
=
train_dataset
.
num_classes
,
ignore_index
=
255
)
# Creat optimizer
num_steps_each_epoch
=
len
(
train_dataset
)
//
args
.
batch_size
...
...
@@ -251,7 +242,7 @@ def main(args):
batch_size
=
args
.
batch_size
,
pretrained_model
=
args
.
pretrained_model
,
save_interval_epochs
=
args
.
save_interval_epochs
,
num_classes
=
args
.
num_classes
,
num_classes
=
train_dataset
.
num_classes
,
num_workers
=
args
.
num_workers
)
...
...
dygraph/val.py
浏览文件 @
6a569885
...
...
@@ -44,22 +44,11 @@ def parse_args():
# params of dataset
parser
.
add_argument
(
'--data_dir'
,
dest
=
'data_dir'
,
help
=
'The root directory of dataset'
,
type
=
str
)
parser
.
add_argument
(
'--val_list'
,
dest
=
'val_list'
,
help
=
'Val list file of dataset'
,
'--dataset'
,
dest
=
'dataset'
,
help
=
'The dataset you want to evaluation'
,
type
=
str
,
default
=
None
)
parser
.
add_argument
(
'--num_classes'
,
dest
=
'num_classes'
,
help
=
'Number of classes'
,
type
=
int
,
default
=
2
)
default
=
'OpticDiscSeg'
)
# params of evaluate
parser
.
add_argument
(
...
...
@@ -140,19 +129,26 @@ def main(args):
places
=
fluid
.
CUDAPlace
(
ParallelEnv
().
dev_id
)
\
if
env_info
[
'place'
]
==
'cuda'
and
fluid
.
is_compiled_with_cuda
()
\
else
fluid
.
CPUPlace
()
if
args
.
dataset
.
lower
()
==
'opticdiscseg'
:
dataset
=
OpticDiscSeg
else
:
raise
Exception
(
"The --dataset set wrong. It should be one of ('OpticDiscSeg',)"
)
with
fluid
.
dygraph
.
guard
(
places
):
eval_transforms
=
T
.
Compose
([
T
.
Resize
(
args
.
input_size
),
T
.
Normalize
()])
eval_dataset
=
OpticDiscSeg
(
transforms
=
eval_transforms
,
mode
=
'eval'
)
eval_dataset
=
dataset
(
transforms
=
eval_transforms
,
mode
=
'eval'
)
if
args
.
model_name
==
'UNet'
:
model
=
models
.
UNet
(
num_classes
=
args
.
num_classes
)
model
=
models
.
UNet
(
num_classes
=
eval_dataset
.
num_classes
)
evaluate
(
model
,
eval_dataset
,
places
=
places
,
model_dir
=
args
.
model_dir
,
num_classes
=
args
.
num_classes
,
num_classes
=
eval_dataset
.
num_classes
,
batch_size
=
args
.
batch_size
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录