[6d389a]: / configs / _base_ / models / bsn_pem.py

Download this file

14 lines (13 with data), 334 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
# model settings
model = dict(
type='PEM',
pem_feat_dim=32,
pem_hidden_dim=256,
pem_u_ratio_m=1,
pem_u_ratio_l=2,
pem_high_temporal_iou_threshold=0.6,
pem_low_temporal_iou_threshold=0.2,
soft_nms_alpha=0.75,
soft_nms_low_threshold=0.65,
soft_nms_high_threshold=0.9,
post_process_top_k=100)