parent
b5ab4b69bc
commit
3a0919bab3
@ -0,0 +1,23 @@
|
||||
from paddle.trainer_config_helpers import *
|
||||
|
||||
settings(batch_size=1000, learning_rate=1e-5)
|
||||
|
||||
input_loc = data_layer(name='input_loc', size=16, height=16, width=1)
|
||||
|
||||
input_conf = data_layer(name='input_conf', size=8, height=1, width=8)
|
||||
|
||||
priorbox = data_layer(name='priorbox', size=32, height=4, width=8)
|
||||
|
||||
detout = detection_output_layer(
|
||||
input_loc=input_loc,
|
||||
input_conf=input_conf,
|
||||
priorbox=priorbox,
|
||||
num_classes=21,
|
||||
nms_threshold=0.45,
|
||||
nms_top_k=400,
|
||||
keep_top_k=200,
|
||||
confidence_threshold=0.01,
|
||||
background_id=0,
|
||||
name='test_detection_output')
|
||||
|
||||
outputs(detout)
|
@ -0,0 +1,25 @@
|
||||
from paddle.trainer_config_helpers import *
|
||||
|
||||
settings(batch_size=1000, learning_rate=1e-5)
|
||||
|
||||
input_loc = data_layer(name='input_loc', size=16, height=16, width=1)
|
||||
|
||||
input_conf = data_layer(name='input_conf', size=8, height=1, width=8)
|
||||
|
||||
priorbox = data_layer(name='priorbox', size=32, height=4, width=8)
|
||||
|
||||
label = data_layer(name='label', size=24, height=4, width=6)
|
||||
|
||||
multibox_loss = multibox_loss_layer(
|
||||
input_loc=input_loc,
|
||||
input_conf=input_conf,
|
||||
priorbox=priorbox,
|
||||
label=label,
|
||||
num_classes=21,
|
||||
overlap_threshold=0.5,
|
||||
neg_pos_ratio=3.0,
|
||||
neg_overlap=0.5,
|
||||
background_id=0,
|
||||
name='test_multibox_loss')
|
||||
|
||||
outputs(multibox_loss)
|
Loading…
Reference in new issue