Skip to content

GitLab

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: ...@@ -34,7 +34,8 @@ MODEL:
DIR_LIMIT_OFFSET: 0.0 DIR_LIMIT_OFFSET: 0.0
NUM_DIR_BINS: 2 NUM_DIR_BINS: 2
USE_MULTI_HEAD: True USE_MULTIHEAD: True
SEPERATE_MULTIHEAD: True
ANCHOR_GENERATOR_CONFIG: [ ANCHOR_GENERATOR_CONFIG: [
{ {
'class_name': 'Car', 'class_name': 'Car',
...@@ -52,7 +53,7 @@ MODEL: ...@@ -52,7 +53,7 @@ MODEL:
'anchor_rotations': [0, 1.57], 'anchor_rotations': [0, 1.57],
'anchor_bottom_heights': [-1.6], 'anchor_bottom_heights': [-1.6],
'align_center': False, 'align_center': False,
'feature_map_stride': 4, 'feature_map_stride': 8,
'matched_threshold': 0.5, 'matched_threshold': 0.5,
'unmatched_threshold': 0.35 'unmatched_threshold': 0.35
}, },
...@@ -62,36 +63,38 @@ MODEL: ...@@ -62,36 +63,38 @@ MODEL:
'anchor_rotations': [0, 1.57], 'anchor_rotations': [0, 1.57],
'anchor_bottom_heights': [-1.6], 'anchor_bottom_heights': [-1.6],
'align_center': False, 'align_center': False,
'feature_map_stride': 4, 'feature_map_stride': 8,
'matched_threshold': 0.5, 'matched_threshold': 0.5,
'unmatched_threshold': 0.35 'unmatched_threshold': 0.35
} }
] ]
SHARED_CONV_NUM_FILTER: 64
RPN_HEAD_CFGS: [ RPN_HEAD_CFGS: [
{ {
'HEAD_CLS_NAME': ['Car'], 'HEAD_CLS_NAME': ['Car'],
'LAYER_NUMS': [1], 'LAYER_NUMS': [],
'LAYER_STRIDES': [1], 'LAYER_STRIDES': [],
'NUM_FILTERS': [512], 'NUM_FILTERS': [],
'UPSAMPLE_STRIDES': [1], 'UPSAMPLE_STRIDES': [],
'NUM_UPSAMPLE_FILTERS': [512] 'NUM_UPSAMPLE_FILTERS': []
}, },
{ {
'HEAD_CLS_NAME': ['Pedestrian'], 'HEAD_CLS_NAME': ['Pedestrian'],
'LAYER_NUMS': [1], 'LAYER_NUMS': [],
'LAYER_STRIDES': [1], 'LAYER_STRIDES': [],
'NUM_FILTERS': [512], 'NUM_FILTERS': [],
'UPSAMPLE_STRIDES': [2], 'UPSAMPLE_STRIDES': [],
'NUM_UPSAMPLE_FILTERS': [512] 'NUM_UPSAMPLE_FILTERS': []
}, },
{ {
'HEAD_CLS_NAME': ['Cyclist'], 'HEAD_CLS_NAME': ['Cyclist'],
'LAYER_NUMS': [1], 'LAYER_NUMS': [],
'LAYER_STRIDES': [1], 'LAYER_STRIDES': [],
'NUM_FILTERS': [512], 'NUM_FILTERS': [],
'UPSAMPLE_STRIDES': [2], 'UPSAMPLE_STRIDES': [],
'NUM_UPSAMPLE_FILTERS': [512] '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