Skip to content

GitLab

"lib/bindings/python/vscode:/vscode.git/clone" did not exist on "99cd9d85a943de0eba06fba89c0e7b5311ea45bb"
Commit 3e92d169 authored by Gus-Guo's avatar Gus-Guo Committed by Shaoshuai Shi
Browse files

update config of multihead

parent b16e0891
Hide whitespace changes
Inline Side-by-side
Showing with 21 additions and 18 deletions
tools/cfgs/kitti_models/second_multihead.yaml
View file @ 3e92d169
......@@ -34,7 +34,8 @@ MODEL:
DIR_LIMIT_OFFSET: 0.0
NUM_DIR_BINS: 2
USE_MULTI_HEAD: True
USE_MULTIHEAD: True
SEPERATE_MULTIHEAD: True
ANCHOR_GENERATOR_CONFIG: [
{
'class_name': 'Car',
......@@ -52,7 +53,7 @@ MODEL:
'anchor_rotations': [0, 1.57],
'anchor_bottom_heights': [-1.6],
'align_center': False,
'feature_map_stride': 4,
'feature_map_stride': 8,
'matched_threshold': 0.5,
'unmatched_threshold': 0.35
},
......@@ -62,36 +63,38 @@ MODEL:
'anchor_rotations': [0, 1.57],
'anchor_bottom_heights': [-1.6],
'align_center': False,
'feature_map_stride': 4,
'feature_map_stride': 8,
'matched_threshold': 0.5,
'unmatched_threshold': 0.35
}
]
SHARED_CONV_NUM_FILTER: 64
RPN_HEAD_CFGS: [
{
'HEAD_CLS_NAME': ['Car'],
'LAYER_NUMS': [1],
'LAYER_STRIDES': [1],
'NUM_FILTERS': [512],
'UPSAMPLE_STRIDES': [1],
'NUM_UPSAMPLE_FILTERS': [512]
'LAYER_NUMS': [],
'LAYER_STRIDES': [],
'NUM_FILTERS': [],
'UPSAMPLE_STRIDES': [],
'NUM_UPSAMPLE_FILTERS': []
},
{
'HEAD_CLS_NAME': ['Pedestrian'],
'LAYER_NUMS': [1],
'LAYER_STRIDES': [1],
'NUM_FILTERS': [512],
'UPSAMPLE_STRIDES': [2],
'NUM_UPSAMPLE_FILTERS': [512]
'LAYER_NUMS': [],
'LAYER_STRIDES': [],
'NUM_FILTERS': [],
'UPSAMPLE_STRIDES': [],
'NUM_UPSAMPLE_FILTERS': []
},
{
'HEAD_CLS_NAME': ['Cyclist'],
'LAYER_NUMS': [1],
'LAYER_STRIDES': [1],
'NUM_FILTERS': [512],
'UPSAMPLE_STRIDES': [2],
'NUM_UPSAMPLE_FILTERS': [512]
'LAYER_NUMS': [],
'LAYER_STRIDES': [],
'NUM_FILTERS': [],
'UPSAMPLE_STRIDES': [],
'NUM_UPSAMPLE_FILTERS': []
}
]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment