Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
433e325b
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
286
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看板
提交
433e325b
编写于
10月 21, 2019
作者:
W
wuyefeilin
提交者:
wuzewu
10月 21, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update loss_select.md and model_builder.py (#67)
* update model_builder.py * update loss_select.md
上级
411b8fce
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
4 addition
and
2 deletion
+4
-2
docs/loss_select.md
docs/loss_select.md
+2
-2
pdseg/models/model_builder.py
pdseg/models/model_builder.py
+2
-0
未找到文件。
docs/loss_select.md
浏览文件 @
433e325b
...
...
@@ -55,13 +55,13 @@ python pdseg/check.py --cfg ./configs/deepglobe_road_extraction.yaml
*
训练
```
shell
python pdseg/train.py
--cfg
./configs/deepglobe_road_extraction.yaml
--use_gpu
SOLVER.LOSS
[
'dice_loss'
,
'bce_loss'
]
python pdseg/train.py
--cfg
./configs/deepglobe_road_extraction.yaml
--use_gpu
SOLVER.LOSS
"['dice_loss','bce_loss']"
```
*
评估
```
python pdseg/eval.py --cfg ./configs/deepglobe_road_extraction.yaml --use_gpu SOLVER.LOSS
['dice_loss','bce_loss']
python pdseg/eval.py --cfg ./configs/deepglobe_road_extraction.yaml --use_gpu SOLVER.LOSS
"['dice_loss','bce_loss']"
```
...
...
pdseg/models/model_builder.py
浏览文件 @
433e325b
...
...
@@ -158,6 +158,8 @@ def build_model(main_prog, start_prog, phase=ModelPhase.TRAIN):
model_func
=
get_func
(
"modeling."
+
model_name
)
loss_type
=
cfg
.
SOLVER
.
LOSS
if
not
isinstance
(
loss_type
,
list
):
loss_type
=
list
(
loss_type
)
if
class_num
>
2
and
((
"dice_loss"
in
loss_type
)
or
(
"bce_loss"
in
loss_type
)):
raise
Exception
(
"dice loss and bce loss is only applicable to binary classfication"
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录