forked from PaddlePaddle/PaddleVideo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfigs.yaml
61 lines (56 loc) · 1.35 KB
/
configs.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
COMMON:
fps: 5
use_gpu: True
label_dic: './configs/index_label_football_8.json'
# debug
PCM_ONLY: False
DEBUG: False
BMN_ONLY: False
LSTM_ONLY: False
PPTSM:
name: "PPTSM"
model_file: "../checkpoints/ppTSM/ppTSM.pdmodel"
params_file: "../checkpoints/ppTSM/ppTSM.pdiparams"
gpu_mem: 8000
device_id: 0
seg_num: 8
seglen: 1
short_size: 256
target_size: 224
batch_size: 32
image_mean: [0.485, 0.456, 0.406]
image_std: [0.229, 0.224, 0.225]
reader_threads: 12
buf_size: 1024
AUDIO:
name: "AUDIO"
model_file: "../checkpoints/AUDIO/__model__"
params_file: "../checkpoints/AUDIO/__param__"
gpu_mem: 8000
device_id: 0
sample_rate: 16000
batch_size: 32
BMN:
name: "BMN"
model_file: "../checkpoints/BMN/__model__"
params_file: "../checkpoints/BMN/__param__"
gpu_mem: 8000
device_id: 0
window_step: 200 # 200
tscale: 200
dscale: 200
batch_size: 8 # 8
nms_thread: 0.7
score_thread: 0.03 # 0.05
ACTION:
name: "ACTION"
model_file: "../checkpoints/LSTM/__model__"
params_file: "../checkpoints/LSTM/__param__"
gpu_mem: 8000
device_id: 0
batch_size: 32
topk: 1
nms_thread: 0.01
nms_offset: 10
classify_score_thread: 0.05 # 0.15
iou_score_thread: 0.1 # 0.4