Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
hapi
提交
5fc538d5
H
hapi
项目概览
PaddlePaddle
/
hapi
通知
11
Star
2
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
4
列表
看板
标记
里程碑
合并请求
7
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
H
hapi
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
4
Issue
4
列表
看板
标记
里程碑
合并请求
7
合并请求
7
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
5fc538d5
编写于
4月 29, 2020
作者:
H
huangjun12
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
refine README and details of bmn model
上级
ed3c3599
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
40 addition
and
20 deletion
+40
-20
examples/bmn/README.md
examples/bmn/README.md
+20
-2
examples/bmn/bmn.yaml
examples/bmn/bmn.yaml
+4
-5
examples/bmn/eval.py
examples/bmn/eval.py
+5
-5
examples/bmn/predict.py
examples/bmn/predict.py
+10
-7
examples/bmn/train.py
examples/bmn/train.py
+1
-1
未找到文件。
examples/bmn/README.md
浏览文件 @
5fc538d5
...
@@ -4,7 +4,7 @@
...
@@ -4,7 +4,7 @@
## 内容
## 内容
-
[
模型简介
](
#模型简介
)
-
[
模型简介
](
#模型简介
)
-
[
代码
结构
](
#代码结构
)
-
[
代码
获取
](
#代码获取
)
-
[
数据准备
](
#数据准备
)
-
[
数据准备
](
#数据准备
)
-
[
模型训练
](
#模型训练
)
-
[
模型训练
](
#模型训练
)
-
[
模型评估
](
#模型评估
)
-
[
模型评估
](
#模型评估
)
...
@@ -22,7 +22,21 @@ BMN Overview
...
@@ -22,7 +22,21 @@ BMN Overview
</p>
</p>
## 代码结构
## 代码获取
### 代码下载及环境变量设置
克隆代码库到本地,并设置`PYTHONPATH`环境变量
```bash
git clone https://github.com/PaddlePaddle/hapi
cd hapi
export PYTHONPATH=`pwd`:$PYTHONPATH
cd examples/bmn
```
### 代码结构
```
```
├── bmn.yaml # 网络配置文件,快速配置参数
├── bmn.yaml # 网络配置文件,快速配置参数
├── run.sh # 快速运行脚本,可直接开始多卡训练
├── run.sh # 快速运行脚本,可直接开始多卡训练
...
@@ -74,6 +88,8 @@ BMN的训练数据采用ActivityNet1.3提供的数据集,我们提供了处理
...
@@ -74,6 +88,8 @@ BMN的训练数据采用ActivityNet1.3提供的数据集,我们提供了处理
-
上述程序会将运行结果保存在
`--output_path`
参数指定的文件夹下,默认为output/EVAL/BMN
\_
results;测试结果保存在
`--result_path`
参数指定的文件夹下,默认为evaluate
\_
results。
-
上述程序会将运行结果保存在
`--output_path`
参数指定的文件夹下,默认为output/EVAL/BMN
\_
results;测试结果保存在
`--result_path`
参数指定的文件夹下,默认为evaluate
\_
results。
-
暂不支持多卡评估。
-
注:评估时可能会出现loss为nan的情况。这是由于评估时用的是单个样本,可能存在没有iou>0.6的样本,所以为nan,对最终的评估结果没有影响。
-
注:评估时可能会出现loss为nan的情况。这是由于评估时用的是单个样本,可能存在没有iou>0.6的样本,所以为nan,对最终的评估结果没有影响。
...
@@ -108,6 +124,8 @@ BMN的训练数据采用ActivityNet1.3提供的数据集,我们提供了处理
...
@@ -108,6 +124,8 @@ BMN的训练数据采用ActivityNet1.3提供的数据集,我们提供了处理
-
上述程序会将运行结果保存在
`--output_path`
参数指定的文件夹下,默认为output/INFER/BMN
\_
results;测试结果保存在
`--result_path`
参数指定的文件夹下,默认为predict
\_
results。
-
上述程序会将运行结果保存在
`--output_path`
参数指定的文件夹下,默认为output/INFER/BMN
\_
results;测试结果保存在
`--result_path`
参数指定的文件夹下,默认为predict
\_
results。
-
暂不支持多卡预测。
## 参考论文
## 参考论文
...
...
examples/bmn/bmn.yaml
浏览文件 @
5fc538d5
...
@@ -15,7 +15,6 @@ TRAIN:
...
@@ -15,7 +15,6 @@ TRAIN:
batch_size
:
4
batch_size
:
4
num_workers
:
4
num_workers
:
4
use_shuffle
:
True
use_shuffle
:
True
device
:
"
gpu"
learning_rate
:
0.001
learning_rate
:
0.001
learning_rate_decay
:
0.1
learning_rate_decay
:
0.1
lr_decay_iter
:
4200
lr_decay_iter
:
4200
...
@@ -28,14 +27,14 @@ TEST:
...
@@ -28,14 +27,14 @@ TEST:
subset
:
"
validation"
subset
:
"
validation"
batch_size
:
1
batch_size
:
1
num_workers
:
1
num_workers
:
1
output_path
:
"
output/EVAL/BMN_results"
output_path
:
"
./
output/EVAL/BMN_results"
result_path
:
"
evaluate_results"
result_path
:
"
./
evaluate_results"
INFER
:
INFER
:
subset
:
"
test"
subset
:
"
test"
batch_size
:
1
batch_size
:
1
num_workers
:
1
num_workers
:
1
filelist
:
'
./infer.list'
filelist
:
'
./infer.list'
output_path
:
"
output/INFER/BMN_results"
output_path
:
"
./
output/INFER/BMN_results"
result_path
:
"
predict_results"
result_path
:
"
./
predict_results"
examples/bmn/eval.py
浏览文件 @
5fc538d5
...
@@ -54,18 +54,18 @@ def parse_args():
...
@@ -54,18 +54,18 @@ def parse_args():
'--weights'
,
'--weights'
,
type
=
str
,
type
=
str
,
default
=
None
,
default
=
None
,
help
=
'weight path
,
None to automatically download weights provided by Paddle.'
help
=
'weight path
.
None to automatically download weights provided by Paddle.'
)
)
parser
.
add_argument
(
parser
.
add_argument
(
'--output_path'
,
'--output_path'
,
type
=
str
,
type
=
str
,
default
=
"output/EVAL/BMN_results"
,
default
=
None
,
help
=
'output dir path
, default to use output/EVAL/BMN_results
'
)
help
=
'output dir path
. None to use config file setting.
'
)
parser
.
add_argument
(
parser
.
add_argument
(
'--result_path'
,
'--result_path'
,
type
=
str
,
type
=
str
,
default
=
"evaluate_results/"
,
default
=
None
,
help
=
'output dir path after post processing
, default to use ./evaluate_results/
'
help
=
'output dir path after post processing
. None to use config file setting.
'
)
)
parser
.
add_argument
(
parser
.
add_argument
(
'--log_interval'
,
'--log_interval'
,
...
...
examples/bmn/predict.py
浏览文件 @
5fc538d5
...
@@ -46,7 +46,10 @@ def parse_args():
...
@@ -46,7 +46,10 @@ def parse_args():
default
=
'bmn.yaml'
,
default
=
'bmn.yaml'
,
help
=
'path to config file of model'
)
help
=
'path to config file of model'
)
parser
.
add_argument
(
parser
.
add_argument
(
'--device'
,
type
=
str
,
default
=
'GPU'
,
help
=
'default use gpu.'
)
'--device'
,
type
=
str
,
default
=
'gpu'
,
help
=
'gpu or cpu, default use gpu.'
)
parser
.
add_argument
(
parser
.
add_argument
(
'--weights'
,
'--weights'
,
type
=
str
,
type
=
str
,
...
@@ -56,18 +59,18 @@ def parse_args():
...
@@ -56,18 +59,18 @@ def parse_args():
parser
.
add_argument
(
parser
.
add_argument
(
'--filelist'
,
'--filelist'
,
type
=
str
,
type
=
str
,
default
=
"infer.list"
,
default
=
None
,
help
=
'infer file list,
default to use ./infer.list
'
)
help
=
'infer file list,
None to use config file setting.
'
)
parser
.
add_argument
(
parser
.
add_argument
(
'--output_path'
,
'--output_path'
,
type
=
str
,
type
=
str
,
default
=
"output/INFER/BMN_results"
,
default
=
None
,
help
=
'output dir path,
default to use output/INFER/BMN_results
'
)
help
=
'output dir path,
None to use config file setting.
'
)
parser
.
add_argument
(
parser
.
add_argument
(
'--result_path'
,
'--result_path'
,
type
=
str
,
type
=
str
,
default
=
"predict_results/"
,
default
=
None
,
help
=
'output dir path after post processing,
default to use ./predict_results/
'
help
=
'output dir path after post processing,
None to use config file setting.
'
)
)
parser
.
add_argument
(
parser
.
add_argument
(
'--log_interval'
,
'--log_interval'
,
...
...
examples/bmn/train.py
浏览文件 @
5fc538d5
...
@@ -49,7 +49,7 @@ def parse_args():
...
@@ -49,7 +49,7 @@ def parse_args():
parser
.
add_argument
(
parser
.
add_argument
(
'--learning_rate'
,
'--learning_rate'
,
type
=
float
,
type
=
float
,
default
=
0.001
,
default
=
None
,
help
=
'learning rate use for training. None to use config file setting.'
)
help
=
'learning rate use for training. None to use config file setting.'
)
parser
.
add_argument
(
parser
.
add_argument
(
'--resume'
,
'--resume'
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录